Spaces:

JuanMa360
/

DD360-Bot

Running

JuanMa360 commited on Mar 16, 2024

Commit

7febbe7

1 Parent(s): b1e9dee

refactor: trus code

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ import gradio as gr
 model = AutoModelForCausalLM.from_pretrained(
     "MILVLG/imp-v1-3b",
     torch_dtype=torch.float32,
-    device_map="auto")
 tokenizer = AutoTokenizer.from_pretrained("tokenizer-dd360",torch_dtype=torch.float32)

 model = AutoModelForCausalLM.from_pretrained(
     "MILVLG/imp-v1-3b",
     torch_dtype=torch.float32,
+    device_map="auto",
+    trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("tokenizer-dd360",torch_dtype=torch.float32)