Spaces:
Running
on
Zero
Running
on
Zero
PommesPeter
commited on
Commit
•
718eae1
1
Parent(s):
b74c6f3
Update app.py
Browse files
app.py
CHANGED
@@ -23,6 +23,9 @@ import models
|
|
23 |
from PIL import Image
|
24 |
from lumina_t2i.transport import create_transport, Sampler
|
25 |
|
|
|
|
|
|
|
26 |
description = """
|
27 |
# Lumina Next Text-to-Image
|
28 |
|
|
|
23 |
from PIL import Image
|
24 |
from lumina_t2i.transport import create_transport, Sampler
|
25 |
|
26 |
+
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
27 |
+
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
28 |
+
|
29 |
description = """
|
30 |
# Lumina Next Text-to-Image
|
31 |
|