declare-lab
/

flacuna-13b-v1.0

Text Generation

text-generation-inference

Model card Files Files and versions

deepanway commited on Jul 4, 2023

Commit

97a4153

·

1 Parent(s): 7124617

Update flacuna.py

Files changed (1) hide show

flacuna.py +1 -1

flacuna.py CHANGED Viewed

@@ -31,7 +31,7 @@ if __name__ == "__main__":
     )
     model = get_peft_model(model, lora_config)
-    weight = torch.load("flacuna-13b-v1.0/pytorch_model.bin", map_location="cpu")
     model.load_state_dict(weight)
     prompt = (

     )
     model = get_peft_model(model, lora_config)
+    weight = torch.load("pytorch_model.bin", map_location="cpu")
     model.load_state_dict(weight)
     prompt = (