chbsaikiran commited on
Commit
6983490
·
1 Parent(s): 0628af7

changed the model file name

Browse files
Files changed (2) hide show
  1. app.py +1 -3
  2. model_bin.pt +3 -0
app.py CHANGED
@@ -17,9 +17,7 @@ model = LlamaForCausalLM(
17
  num_heads=9
18
  )
19
  device = "cpu"
20
-
21
- model_id = "satyanayak/custom-smallmv2135"
22
- checkpoint_path = hf_hub_download(repo_id=model_id, filename="model-dict-step-5500.pt")
23
 
24
  checkpoint = torch.load(checkpoint_path, map_location=device)
25
  model.load_state_dict(checkpoint['model_state_dict'])
 
17
  num_heads=9
18
  )
19
  device = "cpu"
20
+ checkpoint_path = "model_bin.pt"
 
 
21
 
22
  checkpoint = torch.load(checkpoint_path, map_location=device)
23
  model.load_state_dict(checkpoint['model_state_dict'])
model_bin.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a77b73f06315595be13616cb77bf4483955cd9e5c9f40176332feebf7d149e5
3
+ size 704502186