pbevan11
/

llama-3-8b-ocr-correction

Generated from Trainer

4-bit precision

Model card Files Files and versions Community

pbevan11 commited on Jul 26

Commit

c214977

•

1 Parent(s): 9dc4ab2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -118,7 +118,7 @@ First, download the model
 ```python
 from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer
-model_id='pbevan11/llama-3-8b-ocr-correction' # this will be different for you based upon hub_model_id
 model = AutoPeftModelForCausalLM.from_pretrained(model_id).cuda()
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 tokenizer.pad_token = tokenizer.eos_token

 ```python
 from peft import AutoPeftModelForCausalLM
 from transformers import AutoTokenizer
+model_id='pbevan11/llama-3-8b-ocr-correction'
 model = AutoPeftModelForCausalLM.from_pretrained(model_id).cuda()
 tokenizer = AutoTokenizer.from_pretrained(model_id)
 tokenizer.pad_token = tokenizer.eos_token