argilla-warehouse
/

Llama-3.2-1B-Instruct-v2-FC

Text Generation

Generated from Trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

plaguss HF staff commited on Oct 21

Commit

6f6d8fd

•

1 Parent(s): 839a756

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -86,7 +86,7 @@ def parse_response(text: str) -> str | dict[str, any]:
     return text
-model_name_llama = "plaguss/Llama-3.2-1B-Instruct-v2-FC"
 model = AutoModelForCausalLM.from_pretrained(model_name_llama, device_map="auto", torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name_llama)

     return text
+model_name_llama = "argilla-warehouse/Llama-3.2-1B-Instruct-v2-FC"
 model = AutoModelForCausalLM.from_pretrained(model_name_llama, device_map="auto", torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name_llama)