|
--- |
|
license: apache-2.0 |
|
datasets: |
|
- HuggingFaceH4/no_robots |
|
--- |
|
|
|
# Model info |
|
This is Pythia410m-V1-Instruct finetuned on No Robots. This is so it follows system prompts better. |
|
|
|
|
|
``` |
|
from transformers import pipeline |
|
|
|
pipe = pipeline("text-generation", model="SummerSigh/Pythia410m-V1-Instruct-SystemPromptTuning") |
|
|
|
out= pipe("<|im_start|>system\nYou are a good assistant designed to answer all prompts the user asks.<|im_end|><|im_start|>user\nWhat's the meaning of life?<|im_end|><|im_start|>assistant\n",max_length = 500,repetition_penalty = 1.2, temperature = 0.5, do_sample = True) |
|
|
|
print(out[0]["generated_text"]) |
|
|
|
``` |