Sandiago21 commited on
Commit
4e35cec
·
1 Parent(s): e498874

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -104,7 +104,7 @@ tokenizer = LlamaTokenizer.from_pretrained(MODEL_NAME, add_eos_token=True)
104
  tokenizer.pad_token_id = 0
105
 
106
  model = LlamaForCausalLM.from_pretrained(MODEL_NAME, load_in_8bit=True, device_map="auto")
107
- model = PeftModel.from_pretrained(model, "Sandiago21/public-ai-model")
108
  ```
109
 
110
  ### Example of Usage
 
104
  tokenizer.pad_token_id = 0
105
 
106
  model = LlamaForCausalLM.from_pretrained(MODEL_NAME, load_in_8bit=True, device_map="auto")
107
+ model = PeftModel.from_pretrained(model, "Sandiago21/llama-13b-hf")
108
  ```
109
 
110
  ### Example of Usage