up_model / README.md
torchtorchkimtorch's picture
Update README.md
c455f07 verified
์ถ”๋ก  ์‹œ lm-eval-harness ์˜ ARC ๊ธฐ๋ณธ prompt ํ˜•์‹ ์‚ฌ์šฉ (Question: {q}\nAnswer: {a})
[์ถ”๋ก  ์˜ˆ์‹œ]
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
model_id = "torchtorchkimtorch/up_model"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id)
prompt = "Question: What do cells break down to produce energy?\nAnswer:"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=512, do_sample=False)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
```