Spaces:

litagin
/

anime-whisper-demo

Running on Zero

litagin commited on 3 days ago

Commit

7a1d956

•

1 Parent(s): e8dc53b

Change ngram 5

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ generate_kwargs = {
     "language": "Japanese",
     "do_sample": False,
     "num_beams": 1,
-    "no_repeat_ngram_size": 0,
     "max_new_tokens": 64,
 }
@@ -99,8 +99,8 @@ generate_kwargs = {
     "language": "Japanese",
     "do_sample": False,
     "num_beams": 1,
-    "no_repeat_ngram_size": 0,
-    "max_new_tokens": 64,  # 結果が長いときは途中で打ち切る
 }
 ```
 """

     "language": "Japanese",
     "do_sample": False,
     "num_beams": 1,
+    "no_repeat_ngram_size": 5,
     "max_new_tokens": 64,
 }
     "language": "Japanese",
     "do_sample": False,
     "num_beams": 1,
+    "no_repeat_ngram_size": 5,
+    "max_new_tokens": 64,  # 結果が長いときは途中で打ち切られる
 }
 ```
 """