Spaces:
Sleeping
Sleeping
Update src/util/params.py
Browse files- src/util/params.py +2 -2
src/util/params.py
CHANGED
@@ -6,7 +6,7 @@ from diffusers import (
|
|
6 |
AutoencoderKL,
|
7 |
UNet2DConditionModel,
|
8 |
LCMScheduler,
|
9 |
-
|
10 |
StableDiffusionPipeline,
|
11 |
)
|
12 |
|
@@ -43,7 +43,7 @@ text_encoder = CLIPTextModel.from_pretrained(model_path, subfolder="text_encoder
|
|
43 |
if isLCM:
|
44 |
scheduler = LCMScheduler.from_pretrained(model_path, subfolder="scheduler")
|
45 |
else:
|
46 |
-
scheduler =
|
47 |
|
48 |
unet = UNet2DConditionModel.from_pretrained(model_path, subfolder="unet").to(
|
49 |
torch_device
|
|
|
6 |
AutoencoderKL,
|
7 |
UNet2DConditionModel,
|
8 |
LCMScheduler,
|
9 |
+
EulerDiscreteScheduler,
|
10 |
StableDiffusionPipeline,
|
11 |
)
|
12 |
|
|
|
43 |
if isLCM:
|
44 |
scheduler = LCMScheduler.from_pretrained(model_path, subfolder="scheduler")
|
45 |
else:
|
46 |
+
scheduler = EulerDiscreteScheduler.from_pretrained(model_path, subfolder="scheduler")
|
47 |
|
48 |
unet = UNet2DConditionModel.from_pretrained(model_path, subfolder="unet").to(
|
49 |
torch_device
|