Hugging Face's logo

OxxoCodes
/
jamba-small-v1

Text Generation
Transformers
Safetensors
jamba
Inference Endpoints
Model card Files Community
1
OxxoCodes commited on
Commit
a570f75
·
verified ·
1 Parent(s): 9851643

Typo

Browse files
Files changed (1) hide show
  1. prune.py +1 -1
prune.py CHANGED
@@ -37,5 +37,5 @@ def prune_and_copy_additional_layers(original_state_dict):
37
  pruned_state_dict = prune_and_copy_additional_layers(model.state_dict())
38
 
39
  print("Saving weights...")
40
- torch.save(pruned_state_dict, '/scratch/nbrown9/jamba-small-v2.bin')
41
  print("Done!")
 
37
  pruned_state_dict = prune_and_copy_additional_layers(model.state_dict())
38
 
39
  print("Saving weights...")
40
+ torch.save(pruned_state_dict, '/scratch/nbrown9/jamba-small-v1.bin')
41
  print("Done!")