Spaces:

BlinkDL
/

RWKV-Gradio-1

Running on T4

BlinkDL commited on Jun 26, 2023

Commit

f0fb04d

•

1 Parent(s): 8612056

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -189,7 +189,8 @@ def chat(
         if token == 0:
             next_tokens = pipeline.encode('\n\n')
         all_tokens += next_tokens
         if token not in occurrence:
             occurrence[token] = 1
         else:

         if token == 0:
             next_tokens = pipeline.encode('\n\n')
         all_tokens += next_tokens
+        for xxx in occurrence:
+            occurrence[xxx] *= 0.996
         if token not in occurrence:
             occurrence[token] = 1
         else: