Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
|
@@ -175,10 +175,7 @@ try:
|
|
| 175 |
controlnet=controlnet,
|
| 176 |
torch_dtype=dtype,
|
| 177 |
token=hf_token
|
| 178 |
-
)
|
| 179 |
-
|
| 180 |
-
if device_map is None:
|
| 181 |
-
pipe_controlnet.to(device)
|
| 182 |
|
| 183 |
logger.info(" ✓ ControlNet loaded")
|
| 184 |
|
|
|
|
| 175 |
controlnet=controlnet,
|
| 176 |
torch_dtype=dtype,
|
| 177 |
token=hf_token
|
| 178 |
+
).to(device)
|
|
|
|
|
|
|
|
|
|
| 179 |
|
| 180 |
logger.info(" ✓ ControlNet loaded")
|
| 181 |
|