Update cog.yaml

cog.yaml

@@ -1,49 +1,27 @@
-# Configuration for Cog ⚙️
-
+# Configuration for Cog
 build:
+  # Use a GPU because PULSE relies on a vision tower and GPU acceleration
   gpu: true
   python_version: "3.11"
-
   python_packages:
-    # Torch ortamdan geliyor, tekrar sabitleme!
-    - "torch>=2.0.0"
-    - "torchvision>=0.15.0"
-
-    # Hugging Face ekosistemi
+    # Core libraries
+    - "torch==2.1.2"
+    - "torchvision==0.16.2"
     - "transformers==4.37.2"
-    - "tokenizers>=0.15.0"
-    - "sentencepiece>=0.1.99"
-    - "accelerate>=0.25.0"
-    - "sentence-transformers>=2.6.1"
-
-    # Diffusion (PULSE bağımlılığı için)
-    - "diffusers>=0.29.0"
-
-    # Ek paketler
-    - "peft"
-    - "bitsandbytes"
-    - "shortuuid"
-    - "einops>=0.6.1"
-    - "einops-exts>=0.0.4"
+    - "tokenizers==0.15.1"
+    - "sentencepiece==0.1.99"
+    - "safetensors>=0.4.2"
+    - "accelerate==0.30.0"
+    # Vision and utility libraries
     - "timm==0.6.13"
+    - "pillow>=9.5.0"
+    - "requests>=2.31.0"
     - "numpy"
-    - "scikit-learn>=1.2.2"
-    - "pydantic>=1.10.0"
-    - "markdown2[all]"
-    - "requests"
-    - "httpx>=0.25.0"
-    - "fastapi"
-    - "uvicorn"
-    - "gradio>=4.16.0"
-    - "gradio_client>=0.8.1"
-
-    # protobuf fix
-    - "protobuf==3.20.3"
-
-    # LLaVA
+    - "einops==0.6.1"
+    - "einops-exts==0.0.4"
+    - "huggingface_hub==0.13.4"
+    # LLaVA library pulled from GitHub at a fixed tag
     - "llava @ git+https://github.com/haotian-liu/LLaVA@v1.2.0"
 
-  run:
-    - curl -o /usr/local/bin/pget -L "https://github.com/replicate/pget/releases/download/v0.0.3/pget" && chmod +x /usr/local/bin/pget
-
-predict: "predict.py:Predictor"
+# The predictor defines how to run inference
+predict: "predict.py:Predictor"