Spaces:

pyvene
/

AxBench-ReFT-cr1-16K

Runtime error

frankaging commited on Jan 27

Commit

f9cd90a

1 Parent(s): 558908a

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -212,6 +212,7 @@ def generate(
             }
         ] if steering_list else None, # if steering is not provided, we do not steer.
         "streamer": streamer,
         "do_sample": True
     }

             }
         ] if steering_list else None, # if steering is not provided, we do not steer.
         "streamer": streamer,
+        "repetition_penalty": 1.5,
         "do_sample": True
     }