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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -36,12 +36,13 @@ 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
 
 
36
  BASE_MODEL,
37
  scheduler=scheduler,
38
  torch_dtype=torch.bfloat16,
39
+ low_cpu_mem_usage=True # <-- A脩ADIDO: Previene el colapso OOM de la RAM del sistema
40
  )
41
 
42
+ # ELIMINADO: pipe.to("cuda") (Forzaba todo el modelo a la GPU de golpe)
43
+ # A脩ADIDO: Descarga inteligente entre CPU y GPU para ahorrar VRAM
44
  pipe.enable_model_cpu_offload()
45
+
46
  # Diccionario para controlar el estado de carga perezosa de la LoRA de cosplay
47
  loaded_adapters = {}
48