PommesPeter commited on
Commit
718eae1
1 Parent(s): b74c6f3

Update app.py

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