Pection commited on
Commit
95cdb69
·
verified ·
1 Parent(s): f57ef81

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -3,13 +3,13 @@ language:
3
  - en
4
  - th
5
  library_name: transformers
6
- pipeline_tag: text-generation
7
  base_model:
8
  - meta-llama/Llama-3.2-1B
9
  tags:
10
  - text-generation-inference
11
  widget:
12
- - text: "Once upon a time"
13
  ---
14
 
15
  # LLaMA 3 Fine-Tuned Model
@@ -31,4 +31,4 @@ inputs = tokenizer(prompt, return_tensors="pt")
31
  outputs = model.generate(**inputs, max_new_tokens=50)
32
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
33
 
34
- print(response)
 
3
  - en
4
  - th
5
  library_name: transformers
6
+ pipeline_tag: text2text-generation
7
  base_model:
8
  - meta-llama/Llama-3.2-1B
9
  tags:
10
  - text-generation-inference
11
  widget:
12
+ - text: Once upon a time
13
  ---
14
 
15
  # LLaMA 3 Fine-Tuned Model
 
31
  outputs = model.generate(**inputs, max_new_tokens=50)
32
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
33
 
34
+ print(response)