nvidia
/

Llama-3.1-Minitron-4B-Depth-Base

Model card Files Files and versions Community

srvm commited on Aug 21

Commit

9f317d2

•

1 Parent(s): 92e2f46

Update README

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -53,7 +53,7 @@ import torch
 from transformers import AutoTokenizer, LlamaForCausalLM
 # Load the tokenizer and model
-model_path = "nvidia/Llama3.1-Minitron-4B-Depth-Base"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 device = 'cuda'

 from transformers import AutoTokenizer, LlamaForCausalLM
 # Load the tokenizer and model
+model_path = "nvidia/Llama-3.1-Minitron-4B-Depth-Base"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 device = 'cuda'