Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- requirements.txt +9 -19
requirements.txt
CHANGED
@@ -6,26 +6,16 @@ transformers
|
|
6 |
Pillow
|
7 |
pandas
|
8 |
matplotlib
|
9 |
-
torchmetrics[image,multimodal]
|
10 |
-
open_clip_torch
|
11 |
-
accelerate
|
12 |
-
timm
|
13 |
|
14 |
# Зависимости для ONNX и deepghs моделей
|
15 |
-
onnxruntime # или onnxruntime-gpu
|
16 |
huggingface_hub>=0.14.0
|
17 |
-
# dghs-imgutils
|
18 |
-
# но если функции rgb_encode сложны, лучше их иметь
|
19 |
-
# Для простоты, сначала попробуем реализовать ключевые части без них, если возможно,
|
20 |
-
# или добавим, если это необходимо для точного воспроизведения.
|
21 |
-
# imgutils используется в предоставленном коде для rgb_encode
|
22 |
-
# pip install imgutils --no-deps # Если dghs-imgutils не на PyPI, а imgutils - это оно
|
23 |
-
# Попробуем найти imgutils или реализовать аналог rgb_encode
|
24 |
-
numpy # Уже есть, но важен
|
25 |
-
scikit-image # Может использоваться imgutils
|
26 |
-
opencv-python>=4.6.0 # Может использоваться imgutils
|
27 |
-
# natsort # Для сортировки моделей, если бы мы динамически их загружали
|
28 |
|
29 |
-
|
30 |
-
#
|
31 |
-
imgutils
|
|
|
6 |
Pillow
|
7 |
pandas
|
8 |
matplotlib
|
9 |
+
# torchmetrics[image,multimodal] # Убрали, т.к. CLIPScore считаем вручную
|
10 |
+
open_clip_torch # Правильное имя пакета для open_clip
|
11 |
+
accelerate
|
12 |
+
timm
|
13 |
|
14 |
# Зависимости для ONNX и deepghs моделей
|
15 |
+
onnxruntime # или onnxruntime-gpu
|
16 |
huggingface_hub>=0.14.0
|
17 |
+
imgutils # Пакет для dghs-imgutils, если он так называется на PyPI
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
+
numpy
|
20 |
+
# scikit-image # Не используется напрямую, если imgutils справляется
|
21 |
+
# opencv-python>=4.6.0 # Не используется напрямую, если imgutils справляется
|