Spaces:

st192011
/

Janus_Interface

Sleeping

st192011 commited on Jan 21

Commit

d940402

verified ·

1 Parent(s): 1aa2ed2

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
             )
         text = tokenizer.batch_decode(outputs)[0]
@@ -148,7 +148,7 @@ PRIVATE_DB:
                 temperature=0.1,
                 repetition_penalty=1.05,
                 do_sample=True,
-                use_cache=False
             )
         text = tokenizer.batch_decode(outputs)[0]

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