Bytesculptor07 commited on
Commit
c04584e
·
verified ·
1 Parent(s): 849576a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,8 +17,8 @@ from datetime import datetime
17
  from transformers import CLIPImageProcessor
18
  from huggingface_hub import hf_hub_download
19
  from pynvml import *
20
- nvmlInit()
21
- gpu_h = nvmlDeviceGetHandleByIndex(0)
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
 
24
  ctx_limit = 4000
 
17
  from transformers import CLIPImageProcessor
18
  from huggingface_hub import hf_hub_download
19
  from pynvml import *
20
+ #nvmlInit()
21
+ #gpu_h = nvmlDeviceGetHandleByIndex(0)
22
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
23
 
24
  ctx_limit = 4000