-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcog.yaml
More file actions
32 lines (31 loc) · 711 Bytes
/
Copy pathcog.yaml
File metadata and controls
32 lines (31 loc) · 711 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
build:
gpu: true
cuda: "12.4"
python_version: "3.12"
python_packages:
- certifi==2024.8.30
- charset-normalizer==3.4.0
- filelock==3.16.1
- fsspec==2024.10.0
- huggingface-hub==0.26.3
- idna==3.10
- Jinja2==3.1.4
- MarkupSafe==3.0.2
- mpmath==1.3.0
- networkx==3.4.2
- numpy==2.1.3
- packaging==24.2
- PyYAML==6.0.2
- regex==2024.11.6
- requests==2.32.3
- safetensors==0.4.5
- sympy==1.13.1
- tokenizers==0.20.3
- torch==2.5.1
- tqdm==4.67.1
- transformers==4.46.3
- triton==3.1.0
- typing_extensions==4.12.2
- urllib3==2.2.3
- accelerate==1.1.1
predict: "predict.py:Predictor"