Spaces:

sapthesh
/

deepseekv3

Runtime error

sapthesh commited on 3 days ago

Commit

cd9b4c7

verified ·

1 Parent(s): f3c977f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 model_id = "deepseek-ai/DeepSeek-V3"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto") # device_map="auto" for GPU if available
 def predict(message, history):
     conversation = []

 model_id = "deepseek-ai/DeepSeek-V3"
 tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", trust_remote_code=True) # device_map="auto" for GPU if available, ADD trust_remote_code=True
 def predict(message, history):
     conversation = []