Gerchegg commited on
Commit
0ee9a46
·
verified ·
1 Parent(s): b27a83f

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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