Upload ShukaPipeline

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "runs/llama3_saaras_enhi_v0.9.4",
+  "_name_or_path": "sarvamai/shuka_v1",
   "architectures": [
     "ShukaModel"
   ],
@@ -149,7 +149,6 @@
     "vocab_size": 51865
   },
   "audio_model_id": null,
-  "audio_token_index": 32000,
   "auto_map": {
     "AutoConfig": "shuka_config.ShukaConfig",
     "AutoModel": "shuka_model.ShukaModel"

shuka_model.py

@@ -283,8 +283,8 @@ class ShukaModel(
         state_dict = {
             k: v
             for k, v in state_dict.items()
-            if k in self.keep_params
-            or (k in named_params and named_params[k].requires_grad)
+            if k in self.keep_params or k in named_params
+            # or (k in named_params and named_params[k].requires_grad)
         }
         return state_dict