SummerSigh
commited on
Commit
•
75f4691
1
Parent(s):
5e7fcee
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
|
3 |
+
## Usage:
|
4 |
+
```
|
5 |
+
from transformers import pipeline
|
6 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
7 |
+
|
8 |
+
tokenizer = AutoTokenizer.from_pretrained("SummerSigh/Pythia410m-Instruct-SFT")
|
9 |
+
generator = pipeline('text-generation', model = 'SummerSigh/Pythia410m-Instruct-SFT')
|
10 |
+
|
11 |
+
inpopo = input("Text here: ")
|
12 |
+
|
13 |
+
text = generator("<user>" + inpopo + "<user><kinrel>" , max_length = 200, do_sample=True, top_p = 0.7, temperature = 0.5, repetition_penalty = 1.2, pad_token_id=tokenizer.eos_token_id)
|
14 |
+
generated_text = text[0]["generated_text"]
|
15 |
+
parts = generated_text.split("<kinrel>")
|
16 |
+
cropped_text = "<kinrel>".join(parts[:2]) + "<kinrel>"
|
17 |
+
print(cropped_text)
|
18 |
+
|
19 |
+
```
|