NeoPy commited on
Commit
840e123
·
verified ·
1 Parent(s): bb42a5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, torch_dtype="auto", device_map="auto"
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.")