Spaces:

JeanLima2024
/

Multi-Models

Paused

Jean Lima commited on Jan 5

Commit

e83b189

1 Parent(s): ad22e9f

Fix torch_dtype deprecated warning

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ chat_model = AutoModelForCausalLM.from_pretrained(
     LOCAL_MODEL_NAME,
     token=HF_TOKEN,
     trust_remote_code=True,
-    torch_dtype=torch.float16,  # Economia de memória
     device_map="cpu",
     low_cpu_mem_usage=True
 )

     LOCAL_MODEL_NAME,
     token=HF_TOKEN,
     trust_remote_code=True,
+    dtype=torch.float16,  # Economia de memória (Corrigido de torch_dtype)
     device_map="cpu",
     low_cpu_mem_usage=True
 )