add gpu

app.py

@@ -339,6 +339,11 @@ opt.ckpt = opt.sd_path+opt.ckpt
 seed_everything(opt.seed)
 
 # pdb.set_trace()
+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
+
 
 config = OmegaConf.load(f"{opt.config}")
 if os.path.exists(opt.ckpt):

requirements.txt

@@ -1,6 +1,6 @@
---extra-index-url https://download.pytorch.org/whl/cu113 
-torch==1.11.0
-torchvision==0.12.0
+--extra-index-url https://download.pytorch.org/whl/cu113
+torch
+torchvision
 git+https://github.com/huggingface/diffusers.git
 ftfy
 matplotlib