Spaces:

open-nlp
/

Chris-lab

Sleeping

kz209 commited on Jul 16, 2024

Commit

f3984f1

1 Parent(s): 3b35717

update

Files changed (1) hide show

utils/model.py CHANGED Viewed

@@ -16,7 +16,7 @@ class Model():
             device_map="auto",
         )
-    def gen(self, content, temp=0.0, max_length=500):
         sequences = self.pipeline(
             content,
             max_new_tokens=max_length,

             device_map="auto",
         )
+    def gen(self, content, temp=0.1, max_length=500):
         sequences = self.pipeline(
             content,
             max_new_tokens=max_length,