Added model.bin

convert_safetensors_to_bin.py

@@ -0,0 +1,15 @@
+import torch
+from safetensors.torch import load_file
+from transformers import AutoConfig, AutoModel
+
+# Path to your safetensors file
+safetensors_path = "model.safetensors"
+bin_path = "sagvit.bin"
+
+# Load model weights from safetensors
+state_dict = load_file(safetensors_path)
+
+# Save as pytorch_model.bin
+torch.save(state_dict, bin_path)
+
+print(f"Converted {safetensors_path} to {bin_path}.")

sagvit.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff3f3f603954fd987783ae22320d9e941ddb1e53acb9ce5b341d8911bdf3d100
+size 27102094

test_model_loading.py

@@ -1,11 +1,6 @@
 from transformers import AutoModelForImageClassification, AutoImageProcessor
-from safetensors import safe_open
 
-model = AutoModelForImageClassification.from_pretrained(
-    "shravvvv/SAG-ViT",
-    use_safetensors=True,  # Enable safetensors
-    trust_remote_code=True
-)
-processor = AutoImageProcessor.from_pretrained("shravvvv/SAG-ViT", use_safetensors=True, trust_remote_code=True)
+model = AutoModelForImageClassification.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
+processor = AutoImageProcessor.from_pretrained("shravvvv/SAG-ViT", trust_remote_code=True)
 
 print(model)