Update app.py
Browse files
app.py
CHANGED
@@ -136,11 +136,11 @@ eulera_scheduler = EulerAncestralDiscreteScheduler.from_pretrained("stabilityai/
|
|
136 |
|
137 |
|
138 |
controlnet = ControlNetModel.from_pretrained(
|
139 |
-
"xinsir/controlnet-
|
140 |
torch_dtype=torch.float16
|
141 |
)
|
142 |
controlnet_canny = ControlNetModel.from_pretrained(
|
143 |
-
"xinsir/controlnet-
|
144 |
torch_dtype=torch.float16
|
145 |
)
|
146 |
# when test with other base model, you need to change the vae also.
|
|
|
136 |
|
137 |
|
138 |
controlnet = ControlNetModel.from_pretrained(
|
139 |
+
"xinsir/controlnet-union-sdxl-1.0",
|
140 |
torch_dtype=torch.float16
|
141 |
)
|
142 |
controlnet_canny = ControlNetModel.from_pretrained(
|
143 |
+
"xinsir/controlnet-union-sdxl-1.0",
|
144 |
torch_dtype=torch.float16
|
145 |
)
|
146 |
# when test with other base model, you need to change the vae also.
|