Ashishkr
/

llama2-qrecc

Text Generation

Trained with AutoTrain

Inference Endpoints

Model card Files Files and versions Community

Ashishkr commited on Jan 3, 2024

Commit

897e485

·

1 Parent(s): ce69aab

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -21,9 +21,9 @@ import torch
 import re
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-config = PeftConfig.from_pretrained("Ashishkr/llama2-qrecc-context-resolution")
 model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
-model = PeftModel.from_pretrained(model, "Ashishkr/llama2-qrecc-context-resolution").to(device)
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
 def response_generate(

 import re
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+config = PeftConfig.from_pretrained("Ashishkr/llama2-qrecc")
 model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf")
+model = PeftModel.from_pretrained(model, "Ashishkr/llama2-qrecc").to(device)
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf")
 def response_generate(