Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -152,7 +152,7 @@ Have a conversation with an AI using your reference voice!
|
|
152 |
gr.Info("Loading chat model...")
|
153 |
model_name = "deepseek-ai/DeepSeek-V3"
|
154 |
chat_model_state = AutoModelForCausalLM.from_pretrained(
|
155 |
-
model_name,
|
156 |
)
|
157 |
chat_tokenizer_state = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
158 |
gr.Info("Chat model loaded.")
|
|
|
152 |
gr.Info("Loading chat model...")
|
153 |
model_name = "deepseek-ai/DeepSeek-V3"
|
154 |
chat_model_state = AutoModelForCausalLM.from_pretrained(
|
155 |
+
model_name, device_map="auto"
|
156 |
)
|
157 |
chat_tokenizer_state = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
158 |
gr.Info("Chat model loaded.")
|