SidhaarthMurali commited on
Commit
619e0ab
·
verified ·
1 Parent(s): 7983525

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -15
README.md CHANGED
@@ -15,22 +15,7 @@ licence: license
15
  This model is a fine-tuned version of [meta-llama/Llama-3.2-3B](https://huggingface.co/meta-llama/Llama-3.2-3B) on the [openai/gsm8k](https://huggingface.co/datasets/openai/gsm8k) dataset.
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
- ## Quick start
19
 
20
- ```python
21
- from transformers import pipeline
22
-
23
- question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
24
- generator = pipeline("text-generation", model="YWZBrandon/openai-gsm8k_meta-llama-Llama-3.2-3B_sft_lora", device="cuda")
25
- output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
26
- print(output["generated_text"])
27
- ```
28
-
29
- ## Training procedure
30
-
31
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/yuweiz/ActionEditV1/runs/y1bjjrdl)
32
-
33
- This model was trained with SFT.
34
 
35
  ### Framework versions
36
 
 
15
  This model is a fine-tuned version of [meta-llama/Llama-3.2-3B](https://huggingface.co/meta-llama/Llama-3.2-3B) on the [openai/gsm8k](https://huggingface.co/datasets/openai/gsm8k) dataset.
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
 
18
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
 
20
  ### Framework versions
21