fixing error

main.py

@@ -8,7 +8,7 @@ from PIL import Image
 
 # Import library Hugging Face untuk vision-language/LLM model
 # Pastikan library 'transformers' dan 'peft' sudah ada di requirements.txt
-from transformers import AutoProcessor, AutoModelForVision2Seq
+from transformers import AutoProcessor, AutoModelForPreTraining
 from peft import PeftModel
 
 app = FastAPI(
@@ -47,11 +47,11 @@ def load_model_to_ram():
         print(f"Loading processor dari local directory...")
         processor = AutoProcessor.from_pretrained(".")
 
-        # 2. Load Base Model ke CPU
+      # 2. Load Base Model ke CPU (Menggunakan AutoModelForPreTraining khusus PaliGemma)
         print(f"Loading base model: {BASE_MODEL_NAME} (CPU mode)...")
-        base_model = AutoModelForVision2Seq.from_pretrained(
+        base_model = AutoModelForPreTraining.from_pretrained(
             BASE_MODEL_NAME,
-            torch_dtype=torch.float32, # Gunakan float32 untuk CPU stability
+            torch_dtype=torch.float32, # Wajib float32 di CPU gratisan Hugging Face
             low_cpu_mem_usage=True
         )
 

requirement.txt

@@ -1,8 +0,0 @@
-torch
-torchvision
-transformers
-peft
-fastapi
-uvicorn
-python-multipart
-jinja2

requirements.txt

@@ -1,8 +1,10 @@
+--find-links https://download.pytorch.org/whl/cpu
 torch
 torchvision
-transformers
+transformers>=4.41.0
 peft
 fastapi
 uvicorn
 python-multipart
-jinja2
+jinja2
+pillow