Spaces:

FarReelAILab
/

Machine_Mindset

Runtime error

wangrongsheng commited on Jan 5

Commit

fe7fe2a

•

1 Parent(s): 88b741a

support two models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,8 +38,8 @@ if torch.cuda.is_available():
     tokenizer.use_default_system_prompt = False
     model_id_zh = "FarReelAILab/Machine_Mindset_zh_INTJ"
-    model_zh = AutoModelForCausalLM.from_pretrained(model_id_zh, torch_dtype=torch.float16, device_map="auto")
-    tokenizer_zh = AutoTokenizer.from_pretrained(model_id_zh)
     tokenizer_zh.use_default_system_prompt = False

     tokenizer.use_default_system_prompt = False
     model_id_zh = "FarReelAILab/Machine_Mindset_zh_INTJ"
+    model_zh = AutoModelForCausalLM.from_pretrained(model_id_zh, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True)
+    tokenizer_zh = AutoTokenizer.from_pretrained(model_id_zh, trust_remote_code=True)
     tokenizer_zh.use_default_system_prompt = False