Paper99 commited on
Commit
40600b6
1 Parent(s): cbc3656

Update app.py

Browse files

Thanks to

@hysts

's suggestion.

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -41,6 +41,7 @@ pipe.id_encoder.to(device)
41
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
42
  # pipe.set_adapters(["photomaker"], adapter_weights=[1.0])
43
  pipe.fuse_lora()
 
44
 
45
  @spaces.GPU
46
  def generate_image(upload_images, prompt, negative_prompt, style_name, num_steps, style_strength_ratio, num_outputs, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):
 
41
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
42
  # pipe.set_adapters(["photomaker"], adapter_weights=[1.0])
43
  pipe.fuse_lora()
44
+ pipe.to(device)
45
 
46
  @spaces.GPU
47
  def generate_image(upload_images, prompt, negative_prompt, style_name, num_steps, style_strength_ratio, num_outputs, guidance_scale, seed, progress=gr.Progress(track_tqdm=True)):