XA-vito commited on
Commit
04ef942
·
verified ·
1 Parent(s): 65dd957

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -27,11 +27,14 @@ bnb_config = BitsAndBytesConfig(load_in_8bit=True) # Cargar en 8-bit para reduc
27
 
28
  model = AutoModelForCausalLM.from_pretrained(
29
  MODEL_NAME,
30
- torch_dtype=torch.float32, # Forzar CPU con float32
31
- device_map={"": "cpu"}, # Usar solo CPU
 
32
  token=HF_TOKEN
33
  )
34
 
 
 
35
 
36
 
37
  # Memoria conversacional
 
27
 
28
  model = AutoModelForCausalLM.from_pretrained(
29
  MODEL_NAME,
30
+ torch_dtype=torch.float16, # Usa menos RAM que float32
31
+ low_cpu_mem_usage=True, # Reduce uso de memoria en CPU
32
+ device_map={"": "cpu"},
33
  token=HF_TOKEN
34
  )
35
 
36
+ )
37
+
38
 
39
 
40
  # Memoria conversacional