Text Generation
Transformers
PyTorch
llama
text-generation-inference
Inference Endpoints
deepanway commited on
Commit
97a4153
1 Parent(s): 7124617

Update flacuna.py

Browse files
Files changed (1) hide show
  1. flacuna.py +1 -1
flacuna.py CHANGED
@@ -31,7 +31,7 @@ if __name__ == "__main__":
31
  )
32
  model = get_peft_model(model, lora_config)
33
 
34
- weight = torch.load("flacuna-13b-v1.0/pytorch_model.bin", map_location="cpu")
35
  model.load_state_dict(weight)
36
 
37
  prompt = (
 
31
  )
32
  model = get_peft_model(model, lora_config)
33
 
34
+ weight = torch.load("pytorch_model.bin", map_location="cpu")
35
  model.load_state_dict(weight)
36
 
37
  prompt = (