appinitdev commited on
Commit
dcbeb49
verified
1 Parent(s): a2561f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -35,12 +35,13 @@ print("Inicializando tuber铆a multireferencia QwenImageEditPlusPipeline...")
35
  pipe = QwenImageEditPlusPipeline.from_pretrained(
36
  BASE_MODEL,
37
  scheduler=scheduler,
38
- torch_dtype=torch.bfloat16
 
39
  )
40
 
41
  # Descarga de pesos utilizando la API de Hugging Face e instanciaci贸n en GPU
42
- pipe.to("cuda")
43
-
44
  # Diccionario para controlar el estado de carga perezosa de la LoRA de cosplay
45
  loaded_adapters = {}
46
 
 
35
  pipe = QwenImageEditPlusPipeline.from_pretrained(
36
  BASE_MODEL,
37
  scheduler=scheduler,
38
+ torch_dtype=torch.bfloat16,
39
+ low_cpu_mem_usage=True
40
  )
41
 
42
  # Descarga de pesos utilizando la API de Hugging Face e instanciaci贸n en GPU
43
+ #pipe.to("cuda")
44
+ pipe.enable_model_cpu_offload()
45
  # Diccionario para controlar el estado de carga perezosa de la LoRA de cosplay
46
  loaded_adapters = {}
47