Update app_with_diffusers.py

#15
by p20523810 - opened
Files changed (1) hide show
  1. app_with_diffusers.py +2 -2
app_with_diffusers.py CHANGED
@@ -40,7 +40,7 @@ def resize_img(input_image, max_side=1280, min_side=1024, size=None,
40
  instantir_path = f'./models'
41
 
42
  # load pretrained models
43
- pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16)
44
 
45
  # load adapter
46
  load_adapter_to_pipe(
@@ -51,7 +51,7 @@ load_adapter_to_pipe(
51
 
52
  # load previewer lora
53
  pipe.prepare_previewers(instantir_path)
54
- pipe.scheduler = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', subfolder="scheduler")
55
  lcm_scheduler = LCMSingleStepScheduler.from_config(pipe.scheduler.config)
56
 
57
  # load aggregator weights
 
40
  instantir_path = f'./models'
41
 
42
  # load pretrained models
43
+ pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-0.9', torch_dtype=torch.float16)
44
 
45
  # load adapter
46
  load_adapter_to_pipe(
 
51
 
52
  # load previewer lora
53
  pipe.prepare_previewers(instantir_path)
54
+ pipe.scheduler = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-xl-base-0.9', subfolder="scheduler")
55
  lcm_scheduler = LCMSingleStepScheduler.from_config(pipe.scheduler.config)
56
 
57
  # load aggregator weights