Spaces:

frankrobotics
/

deepseek-ai-DeepSeek-R1

Runtime error

frankrobotics commited on Mar 7

Commit

d8ecca8

verified ·

1 Parent(s): c647451

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,6 +9,7 @@ pipe = pipeline(
     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
 )
 messages = [
     {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},

     model=model_id,
     torch_dtype=torch.bfloat16,
     device_map="auto",
+    trust_remote_code=True,
 )
 messages = [
     {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},