Spaces:
Running
on
Zero
Running
on
Zero
Update app_with_diffusers.py
#15
by
p20523810
- opened
- 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-
|
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-
|
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
|