rafavidal1709 commited on
Commit
5e77219
verified
1 Parent(s): 364da20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -14,11 +14,11 @@ tokenizer = AutoTokenizer.from_pretrained(
14
  model = AutoModelForCausalLM.from_pretrained(
15
  MODEL_NAME,
16
  trust_remote_code=True,
17
- torch_dtype=torch.float32,
18
- device_map="cpu",
19
- low_cpu_mem_usage=True,
20
- load_in_8bit=False,
21
- offload_folder="offload"
22
  )
23
 
24
  def summarize_text(text):
 
14
  model = AutoModelForCausalLM.from_pretrained(
15
  MODEL_NAME,
16
  trust_remote_code=True,
17
+ torch_dtype=torch.float32, # For莽a o uso de float32 em CPU
18
+ device_map="cpu", # Garante que o modelo rode na CPU
19
+ low_cpu_mem_usage=True, # Otimiza uso de mem贸ria
20
+ load_in_8bit=False, # Evita tentativas de quantiza莽茫o FP8
21
+ offload_folder="offload" # Pasta para descarregar partes do modelo
22
  )
23
 
24
  def summarize_text(text):