Maximofn commited on
Commit
264ba49
·
1 Parent(s): 67c95e3

fix(src): :rocket: Remove device_map

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def initialize_model(model):
17
  subfolder="transformer",
18
  quantization_config=quant_config,
19
  torch_dtype=torch.bfloat16,
20
- device_map="balanced",
21
  )
22
 
23
  # Cargar el pipeline
 
17
  subfolder="transformer",
18
  quantization_config=quant_config,
19
  torch_dtype=torch.bfloat16,
20
+ # device_map="balanced",
21
  )
22
 
23
  # Cargar el pipeline