zideliu commited on
Commit
1efe1b6
1 Parent(s): 7da0921

Update app.py

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +0 -3
app.py CHANGED
@@ -18,7 +18,7 @@ from PIL import Image
18
  print("cuda available:",torch.cuda.is_available())
19
  print("cuda device count:",torch.cuda.device_count())
20
  print("cuda device name:",torch.cuda.get_device_name(0))
21
- print(os.system("nvidia-smi"))
22
  print(os.system("nvcc --version"))
23
 
24
  empty_context = np.load("assets/contexts/empty_context.npy")
 
18
  print("cuda available:",torch.cuda.is_available())
19
  print("cuda device count:",torch.cuda.device_count())
20
  print("cuda device name:",torch.cuda.get_device_name(0))
21
+ # print(os.system("nvidia-smi"))
22
  print(os.system("nvcc --version"))
23
 
24
  empty_context = np.load("assets/contexts/empty_context.npy")
requirements.txt CHANGED
@@ -1,5 +1,3 @@
1
- torch==1.13.1
2
- torchvision==0.14.1
3
  accelerate==0.12.0
4
  absl-py
5
  ml_collections
@@ -9,5 +7,4 @@ transformers==4.23.1
9
  loguru
10
  gradio==3.34.0
11
  omegaconf
12
- wget
13
  xformers==0.0.16
 
 
 
1
  accelerate==0.12.0
2
  absl-py
3
  ml_collections
 
7
  loguru
8
  gradio==3.34.0
9
  omegaconf
 
10
  xformers==0.0.16