VOIDER commited on
Commit
bad19ae
·
verified ·
1 Parent(s): 57c45ff

Update requirements.txt

Browse files
Files changed (1) hide show
  1. 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, если GPU доступен и настроен в Space
16
  huggingface_hub>=0.14.0
17
- # dghs-imgutils и hbutils могут быть не нужны напрямую, если мы реализуем их функции
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
- # Уточнение: dghs-imgutils доступен на PyPI как 'imgutils' (скорее всего)
30
- # pip install imgutils
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 справляется