artificiallover0 commited on
Commit
a97d552
·
verified ·
1 Parent(s): 597feb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ else:
18
  torch_dtype = torch.float32
19
 
20
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
21
- pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
22
  pipe.load_lora_weights("artificiallover0/man_asshole")
23
  pipe.fuse_lora()
24
  pipe = pipe.to(device)
 
18
  torch_dtype = torch.float32
19
 
20
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
21
+ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
22
  pipe.load_lora_weights("artificiallover0/man_asshole")
23
  pipe.fuse_lora()
24
  pipe = pipe.to(device)