raw torch load

utils/color_model.py

@@ -1,7 +1,14 @@
 import torch
-from transformers import AutoModel
+from huggingface_hub import hf_hub_download
 
-model = AutoModel.from_pretrained("danielritchie/vibe-color-model")
+# Download model file from HF
+model_path = hf_hub_download(
+    repo_id="danielritchie/vibe-color-model",
+    filename="pytorch_model.bin"
+)
+
+# Load full torch model
+model = torch.load(model_path, map_location=torch.device("cpu"))
 model.eval()