Upload folder using huggingface_hub

app.py

@@ -9,10 +9,13 @@ MODEL_ID = "IFM/K2-Horizon-0.9B-GGUF"
 @spaces.GPU(duration=60)
 def load_model():
     """Load the model and tokenizer with proper config handling."""
-    # Load config first to check model_type
-    config = AutoConfig.from_pretrained(MODEL_ID, trust_remote_code=True)
+    try:
+        config = AutoConfig.from_pretrained(MODEL_ID, trust_remote_code=True)
+    except ValueError:
+        # Model config doesn't have model_type, set it manually
+        config = AutoConfig.for_model("k2_horizon")
+        config.is_encoder_decoder = False
     
-    # If model_type is not set, set it based on the architecture
     if not hasattr(config, "model_type") or config.model_type is None:
         config.model_type = "k2_horizon"