upload

demo.py

@@ -19,6 +19,10 @@ from paintmind.utils.datasets import vae_transforms
 from PIL import Image
 
 transform = vae_transforms('test')
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+# Tesla T4
 
 
 def norm_ip(img, low, high):
@@ -167,4 +171,4 @@ with gr.Blocks() as demo:
 
 # Launch the demo
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()