SummerSigh commited on
Commit
75f4691
1 Parent(s): 5e7fcee

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +19 -0
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
+ ```