randomani commited on
Commit
b6f475f
·
verified ·
1 Parent(s): ce95fe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ pipe = DiffusionPipeline.from_pretrained(
23
  ).to(device)
24
 
25
  # Optional CPU offloading to save some GPU Memory
26
- pipe.enable_model_cpu_offload()
27
 
28
  # # Loading Trained DreamBooth LoRA Weights
29
  pipe.load_lora_weights(model_repo_id)
 
23
  ).to(device)
24
 
25
  # Optional CPU offloading to save some GPU Memory
26
+ # pipe.enable_model_cpu_offload()
27
 
28
  # # Loading Trained DreamBooth LoRA Weights
29
  pipe.load_lora_weights(model_repo_id)