lyu-boxuan
/

llama-3-youko-8b-En-Ja-MT-LoRA

text-generation

machine translation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

lyu-boxuan commited on May 11

Commit

51f6e43

•

1 Parent(s): 6383e73

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -36,7 +36,7 @@ def create_input(text, tokenizer):
     return input_ids
-model_id = "lyu/MT/output/llama3-sft-lora-16-NLLB-100k-run2/merge"
 model = AutoModelForCausalLM.from_pretrained(
     model_id, torch_dtype=torch.bfloat16, attn_implementation="flash_attention_2"
 ).cuda()

     return input_ids
+model_id = "lyu-boxuan/llama-3-youko-8b-En-Ja-MT-LoRA"
 model = AutoModelForCausalLM.from_pretrained(
     model_id, torch_dtype=torch.bfloat16, attn_implementation="flash_attention_2"
 ).cuda()