Update app.py
Browse files
app.py
CHANGED
|
@@ -252,7 +252,7 @@ if torch.cuda.is_available():
|
|
| 252 |
pipe = utils.load_pipeline(MODEL, device)
|
| 253 |
else:
|
| 254 |
logger.warning("CUDA not available, running on CPU")
|
| 255 |
-
pipe =
|
| 256 |
|
| 257 |
# Process styles
|
| 258 |
styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
|
|
|
|
| 252 |
pipe = utils.load_pipeline(MODEL, device)
|
| 253 |
else:
|
| 254 |
logger.warning("CUDA not available, running on CPU")
|
| 255 |
+
pipe = utils.load_pipeline(MODEL, device)
|
| 256 |
|
| 257 |
# Process styles
|
| 258 |
styles = {k["name"]: (k["prompt"], k["negative_prompt"]) for k in style_list}
|