Hugging Face's logo

Spaces:
alexeyGod
/
Test_new_mod
Runtime error

App Files Community
alexeyGod commited on
Commit
e6764a7
·
verified ·
1 Parent(s): 10a4310

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -36,6 +36,7 @@ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
36
  #<lora:RearViewXlV2-CruzFlesh:0.75>, hairy ass, asshole
37
 
38
  #pipe.load_lora_weights("Bear.safetensors")
 
39
  pipe = pipe.to(device)
40
 
41
 
 
36
  #<lora:RearViewXlV2-CruzFlesh:0.75>, hairy ass, asshole
37
 
38
  #pipe.load_lora_weights("Bear.safetensors")
39
+ pipe.enable_model_cpu_offload()
40
  pipe = pipe.to(device)
41
 
42