Portx
/

trial-multi-model

Model card Files Files and versions

Portx commited on Mar 3, 2025

Commit

5499763

·

verified ·

1 Parent(s): 999aaf7

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -14,8 +14,8 @@ run("pip install torch torchvision torchaudio --extra-index-url https://download
 from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig
-#model_id = "ibm-granite/granite-vision-3.2-2b"
-model_id = "Portx/granite-vision-3.1-2b-preview-do-extractor"
 bnb_config = BitsAndBytesConfig(
         load_in_4bit=True,

 from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig
+model_id = "ibm-granite/granite-vision-3.2-2b"
+#model_id = "Portx/granite-vision-3.1-2b-preview-do-extractor"
 bnb_config = BitsAndBytesConfig(
         load_in_4bit=True,