liuyizhang
commited on
Commit
•
76fc69a
1
Parent(s):
7a52d01
update app.py
Browse files- app.py +1 -1
- requirements.txt +4 -3
app.py
CHANGED
@@ -793,7 +793,7 @@ def get_model_device(module):
|
|
793 |
return 'Error'
|
794 |
|
795 |
ram_enable = True
|
796 |
-
kosmos_enable =
|
797 |
|
798 |
task_types = ["detection", "segment", "inpainting"]
|
799 |
if lama_cleaner_enable:
|
|
|
793 |
return 'Error'
|
794 |
|
795 |
ram_enable = True
|
796 |
+
kosmos_enable = False
|
797 |
|
798 |
task_types = ["detection", "segment", "inpainting"]
|
799 |
if lama_cleaner_enable:
|
requirements.txt
CHANGED
@@ -17,15 +17,16 @@ termcolor
|
|
17 |
timm
|
18 |
torch
|
19 |
torchvision
|
20 |
-
|
21 |
-
transformers@git+https://github.com/huggingface/transformers.git@main
|
22 |
yapf
|
23 |
numba
|
24 |
scipy
|
25 |
safetensors
|
26 |
pynvml
|
27 |
sentencepiece
|
|
|
28 |
|
29 |
-
|
30 |
openmim==0.1.5
|
31 |
mmcv==2.0.0
|
|
|
17 |
timm
|
18 |
torch
|
19 |
torchvision
|
20 |
+
transformers==4.27.4
|
21 |
+
# transformers@git+https://github.com/huggingface/transformers.git@main
|
22 |
yapf
|
23 |
numba
|
24 |
scipy
|
25 |
safetensors
|
26 |
pynvml
|
27 |
sentencepiece
|
28 |
+
loguru
|
29 |
|
30 |
+
lama-cleaner==1.2.3
|
31 |
openmim==0.1.5
|
32 |
mmcv==2.0.0
|