Spaces:
Running
on
Zero
Running
on
Zero
Update app_with_diffusers.py
Browse files- 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
|