theblackcat102 commited on
Commit
8034190
1 Parent(s): 59a004f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -0
README.md CHANGED
@@ -1,3 +1,35 @@
1
  ---
2
  license: mit
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ language:
4
+ - en
5
+ - zh
6
+ - de
7
+ - fr
8
+ - ja
9
+ - ko
10
+ pipeline_tag: text2text-generation
11
+ tags:
12
+ - text-generation-inference
13
  ---
14
+
15
+ Generate title for conversation
16
+
17
+ ```python
18
+ model_name = "theblackcat102/alpaca-title-generator-mt0-large"
19
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
20
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
21
+ question = 'Hi\nHow can I help you?'
22
+ encodes = tokenizer(question, return_tensors='pt')
23
+ outputs = model.generate(encodes.input_ids,
24
+ max_length=512,
25
+ do_sample=True,
26
+ repetition_penalty=1.2,
27
+ top_k=50,
28
+ num_return_sequences=1,
29
+ early_stopping=True
30
+ )
31
+ for i, beam_output in enumerate(outputs):
32
+ print('-----')
33
+ print("{}".format(tokenizer.decode(beam_output, skip_special_tokens=True)))
34
+ # > Help requested.
35
+ ```