aixsatoshi commited on
Commit
fcc5976
1 Parent(s): 6baa598

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -4
README.md CHANGED
@@ -55,9 +55,6 @@ def predict(message, history, tokens, temperature, language):
55
  model_inputs,
56
  streamer=streamer,
57
  max_new_tokens=int(tokens),
58
- do_sample=True,
59
- top_p=0.95,
60
- top_k=100,
61
  temperature=float(temperature),
62
  num_beams=1,
63
  stopping_criteria=StoppingCriteriaList([stop])
@@ -77,7 +74,7 @@ demo = gr.ChatInterface(
77
  title="Honyaku-7b webui",
78
  description="Translate using Honyaku-7b model",
79
  additional_inputs=[
80
- gr.Slider(100, 1024, value=1000, label="Tokens"),
81
  gr.Slider(0.0, 1.0, value=0.3, label="Temperature"),
82
  gr.Dropdown(choices=languages, value="Japanese", label="Language")
83
  ]
 
55
  model_inputs,
56
  streamer=streamer,
57
  max_new_tokens=int(tokens),
 
 
 
58
  temperature=float(temperature),
59
  num_beams=1,
60
  stopping_criteria=StoppingCriteriaList([stop])
 
74
  title="Honyaku-7b webui",
75
  description="Translate using Honyaku-7b model",
76
  additional_inputs=[
77
+ gr.Slider(100, 4096, value=1000, label="Tokens"),
78
  gr.Slider(0.0, 1.0, value=0.3, label="Temperature"),
79
  gr.Dropdown(choices=languages, value="Japanese", label="Language")
80
  ]