OxxoCodes
/

jamba-small-v1

Text Generation

Inference Endpoints

Model card Files Files and versions Community

OxxoCodes commited on Apr 9

Commit

a570f75

•

1 Parent(s): 9851643

Typo

Files changed (1) hide show

prune.py +1 -1

prune.py CHANGED Viewed

@@ -37,5 +37,5 @@ def prune_and_copy_additional_layers(original_state_dict):
 pruned_state_dict = prune_and_copy_additional_layers(model.state_dict())
 print("Saving weights...")
-torch.save(pruned_state_dict, '/scratch/nbrown9/jamba-small-v2.bin')
 print("Done!")

 pruned_state_dict = prune_and_copy_additional_layers(model.state_dict())
 print("Saving weights...")
+torch.save(pruned_state_dict, '/scratch/nbrown9/jamba-small-v1.bin')
 print("Done!")