wondervictor commited on
Commit
98a4644
1 Parent(s): 97c3ff1

update env

Browse files
Files changed (2) hide show
  1. app.py +6 -5
  2. requirements.txt +4 -3
app.py CHANGED
@@ -1,14 +1,15 @@
1
  import os
2
- import spaces
3
  from pip._internal import main
4
 
5
  # os.system('python model/segment_anything_2/setup.py build_ext --inplace')
6
 
7
- main(['install', 'timm==1.0.8'])
8
- main(['install', 'samv2'])
9
- main(['install', 'torch==2.1.2'])
10
- main(['install', 'numpy==1.21.6'])
11
 
 
 
 
 
 
 
12
  import timm
13
 
14
  print("installed", timm.__version__)
 
1
  import os
 
2
  from pip._internal import main
3
 
4
  # os.system('python model/segment_anything_2/setup.py build_ext --inplace')
5
 
 
 
 
 
6
 
7
+ # main(['install', 'timm==1.0.8'])
8
+ # main(['install', 'samv2'])
9
+ # main(['install', 'bitsandbytes', '--upgrade'])
10
+ # main(['install', 'torch==2.1.2'])
11
+ # main(['install', 'numpy==1.21.6'])
12
+ import spaces
13
  import timm
14
 
15
  print("installed", timm.__version__)
requirements.txt CHANGED
@@ -16,7 +16,7 @@ tqdm==4.64.1
16
  transformers==4.31.0
17
  uvicorn==0.23.2
18
  scipy==1.11.2
19
- bitsandbytes==0.41.1
20
  blobfile
21
  mypy
22
  pytest
@@ -30,7 +30,8 @@ deepspeed
30
  pycocoevalcap
31
  torchscale
32
  gradio
33
- timm
 
34
  torch
35
  hydra-core
36
- # samv2
 
16
  transformers==4.31.0
17
  uvicorn==0.23.2
18
  scipy==1.11.2
19
+ bitsandbytes
20
  blobfile
21
  mypy
22
  pytest
 
30
  pycocoevalcap
31
  torchscale
32
  gradio
33
+ timm==1.0.8
34
+ samv2
35
  torch
36
  hydra-core
37
+