nirajandhakal
/

LLaMA3-Reasoning

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nirajandhakal commited on Sep 25

Commit

05f15f2

•

1 Parent(s): fabb9f4

truncation=True added to pipeline

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ tokenizer = AutoTokenizer.from_pretrained("nirajandhakal/LLaMA3-Reasoning")
 model = AutoModelForCausalLM.from_pretrained("nirajandhakal/LLaMA3-Reasoning")
-pipe = pipeline("text-generation", model="nirajandhakal/LLaMA3-Reasoning")
 # Define a prompt for the model
 prompt = "What are the benefits of using artificial intelligence in healthcare?"

 model = AutoModelForCausalLM.from_pretrained("nirajandhakal/LLaMA3-Reasoning")
+pipe = pipeline("text-generation", model="nirajandhakal/LLaMA3-Reasoning", truncation=True)
 # Define a prompt for the model
 prompt = "What are the benefits of using artificial intelligence in healthcare?"