Spaces:
Runtime error
Runtime error
update lightweight code
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 27 |
# device = "cpu"
|
| 28 |
|
| 29 |
if device == "cuda":
|
| 30 |
-
args.image_caption_device = "
|
| 31 |
args.dense_caption_device = "cuda"
|
| 32 |
args.semantic_segment_device = "cuda"
|
| 33 |
args.contolnet_device = "cuda"
|
|
|
|
| 27 |
# device = "cpu"
|
| 28 |
|
| 29 |
if device == "cuda":
|
| 30 |
+
args.image_caption_device = "cuda"
|
| 31 |
args.dense_caption_device = "cuda"
|
| 32 |
args.semantic_segment_device = "cuda"
|
| 33 |
args.contolnet_device = "cuda"
|