Upload ShukaPipeline
Browse files- config.json +1 -2
- shuka_model.py +2 -2
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"ShukaModel"
|
5 |
],
|
@@ -149,7 +149,6 @@
|
|
149 |
"vocab_size": 51865
|
150 |
},
|
151 |
"audio_model_id": null,
|
152 |
-
"audio_token_index": 32000,
|
153 |
"auto_map": {
|
154 |
"AutoConfig": "shuka_config.ShukaConfig",
|
155 |
"AutoModel": "shuka_model.ShukaModel"
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "sarvamai/shuka_v1",
|
3 |
"architectures": [
|
4 |
"ShukaModel"
|
5 |
],
|
|
|
149 |
"vocab_size": 51865
|
150 |
},
|
151 |
"audio_model_id": null,
|
|
|
152 |
"auto_map": {
|
153 |
"AutoConfig": "shuka_config.ShukaConfig",
|
154 |
"AutoModel": "shuka_model.ShukaModel"
|
shuka_model.py
CHANGED
@@ -283,8 +283,8 @@ class ShukaModel(
|
|
283 |
state_dict = {
|
284 |
k: v
|
285 |
for k, v in state_dict.items()
|
286 |
-
if k in self.keep_params
|
287 |
-
or (k in named_params and named_params[k].requires_grad)
|
288 |
}
|
289 |
return state_dict
|
290 |
|
|
|
283 |
state_dict = {
|
284 |
k: v
|
285 |
for k, v in state_dict.items()
|
286 |
+
if k in self.keep_params or k in named_params
|
287 |
+
# or (k in named_params and named_params[k].requires_grad)
|
288 |
}
|
289 |
return state_dict
|
290 |
|