partex-nv
/

Llama-3.1-8B-VaaniSetu-EN2PA

text-2-text translation

English2Punjabi

Model card Files Files and versions Community

partex-nv commited on 17 days ago

Commit

27569a5

•

1 Parent(s): ddb32f2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -64,9 +64,9 @@ import torch
 # Load model and tokenizer
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("rohitanurag/Llama-3.1-8B-VaaniSetu-EN2PA", use_auth_token=HF_TOKEN)
     model = AutoModelForCausalLM.from_pretrained(
-        "rohitanurag/Llama-3.1-8B-VaaniSetu-EN2PA",
         torch_dtype=torch.bfloat16,
         device_map="auto",  # Automatically moves model to GPU
     )

 # Load model and tokenizer
 def load_model():
+    tokenizer = AutoTokenizer.from_pretrained("partex-nv/Llama-3.1-8B-VaaniSetu-EN2PA")
     model = AutoModelForCausalLM.from_pretrained(
+        "partex-nv/Llama-3.1-8B-VaaniSetu-EN2PA",
         torch_dtype=torch.bfloat16,
         device_map="auto",  # Automatically moves model to GPU
     )