Spaces:

open-nlp
/

Chris-lab

Sleeping

kz209 commited on Jul 23, 2024

Commit

717032b

•

1 Parent(s): 01a2ce5

update

Files changed (1) hide show

utils/model.py CHANGED Viewed

@@ -13,7 +13,6 @@ class Model():
             model=model_name,
             tokenizer=self.tokenizer,
             torch_dtype=torch.bfloat16,
-            trust_remote_code=True,
             device_map="auto",
         )

             model=model_name,
             tokenizer=self.tokenizer,
             torch_dtype=torch.bfloat16,
             device_map="auto",
         )