Spaces:

st192011
/

Janus_Interface

Sleeping

st192011 commited on Jan 21

Commit

7f5856e

verified ·

1 Parent(s): 7215403

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ RAW NOTE:
                 max_new_tokens=256,
                 temperature=0.1,
                 do_sample=True,
-                use_cache=False  # <--- ADD THIS LINE
             )
         text = tokenizer.batch_decode(outputs)[0]
@@ -147,7 +147,8 @@ PRIVATE_DB:
                 max_new_tokens=1024,
                 temperature=0.1,
                 repetition_penalty=1.05,
-                do_sample=True
             )
         text = tokenizer.batch_decode(outputs)[0]

                 max_new_tokens=256,
                 temperature=0.1,
                 do_sample=True,
+                use_cache=False
             )
         text = tokenizer.batch_decode(outputs)[0]
                 max_new_tokens=1024,
                 temperature=0.1,
                 repetition_penalty=1.05,
+                do_sample=True,
+                use_cache=False
             )
         text = tokenizer.batch_decode(outputs)[0]