CanerDedeoglu commited on
Commit
1308d4b
·
verified ·
1 Parent(s): 76bd29f

Update cog.yaml

Browse files
Files changed (1) hide show
  1. cog.yaml +26 -17
cog.yaml CHANGED
@@ -1,40 +1,49 @@
1
  # Configuration for Cog ⚙️
2
- # Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md
3
 
4
  build:
5
  gpu: true
6
-
7
  python_version: "3.11"
8
 
9
  python_packages:
10
- - "torch==2.1.2"
11
- - "torchvision==0.16.2"
 
 
 
12
  - "transformers==4.37.2"
13
- - "tokenizers==0.15.1"
14
- - "sentencepiece==0.1.99"
15
- - "accelerate==0.34.2"
16
- - "sentence-transformers==2.6.1"
17
- - "diffusers==0.29.2"
 
 
 
 
18
  - "peft"
19
  - "bitsandbytes"
20
  - "shortuuid"
21
- - "einops==0.6.1"
22
- - "einops-exts==0.0.4"
23
  - "timm==0.6.13"
24
  - "numpy"
25
- - "scikit-learn==1.2.2"
26
- - "pydantic"
27
  - "markdown2[all]"
28
  - "requests"
29
- - "httpx==0.24.0"
30
  - "fastapi"
31
  - "uvicorn"
32
- - "gradio==4.16.0"
33
- - "gradio_client==0.8.1"
 
 
 
 
 
34
  - "llava @ git+https://github.com/haotian-liu/LLaVA@v1.2.0"
35
 
36
  run:
37
  - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.0.3/pget" && chmod +x /usr/local/bin/pget
38
 
39
- # predict.py defines how predictions are run on your model
40
  predict: "predict.py:Predictor"
 
1
  # Configuration for Cog ⚙️
 
2
 
3
  build:
4
  gpu: true
 
5
  python_version: "3.11"
6
 
7
  python_packages:
8
+ # Torch ortamdan geliyor, tekrar sabitleme!
9
+ - "torch>=2.0.0"
10
+ - "torchvision>=0.15.0"
11
+
12
+ # Hugging Face ekosistemi
13
  - "transformers==4.37.2"
14
+ - "tokenizers>=0.15.0"
15
+ - "sentencepiece>=0.1.99"
16
+ - "accelerate>=0.25.0"
17
+ - "sentence-transformers>=2.6.1"
18
+
19
+ # Diffusion (PULSE bağımlılığı için)
20
+ - "diffusers>=0.29.0"
21
+
22
+ # Ek paketler
23
  - "peft"
24
  - "bitsandbytes"
25
  - "shortuuid"
26
+ - "einops>=0.6.1"
27
+ - "einops-exts>=0.0.4"
28
  - "timm==0.6.13"
29
  - "numpy"
30
+ - "scikit-learn>=1.2.2"
31
+ - "pydantic>=1.10.0"
32
  - "markdown2[all]"
33
  - "requests"
34
+ - "httpx>=0.25.0"
35
  - "fastapi"
36
  - "uvicorn"
37
+ - "gradio>=4.16.0"
38
+ - "gradio_client>=0.8.1"
39
+
40
+ # protobuf fix
41
+ - "protobuf==3.20.3"
42
+
43
+ # LLaVA
44
  - "llava @ git+https://github.com/haotian-liu/LLaVA@v1.2.0"
45
 
46
  run:
47
  - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.0.3/pget" && chmod +x /usr/local/bin/pget
48
 
 
49
  predict: "predict.py:Predictor"