FredZhang7 commited on
Commit
b1182d2
1 Parent(s): 71aeaca

Update widget

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -4,8 +4,8 @@ language:
4
  - en
5
  widget:
6
  - text: 1girl, fate
7
- - text: 1boy, league of legends
8
- - text: 1girl, genshin impact
9
  - text: 1boy, national basketball association
10
  - text: 1girl, spy x
11
  - text: 1girl, absurdres
@@ -56,7 +56,7 @@ prompt = r'1girl, genshin'
56
  nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
57
 
58
  # generate 10 samples using greedy search
59
- outs = nlp(prompt, max_length=76, num_return_sequences=10, do_sample=True, repetition_penalty=1.1, temperature=0.7, top_k=4, early_stopping=True)
60
 
61
  print('\nInput:\n' + 100 * '-')
62
  print('\033[96m' + prompt + '\033[0m')
 
4
  - en
5
  widget:
6
  - text: 1girl, fate
7
+ - text: 1boy, league of
8
+ - text: 1girl, genshin
9
  - text: 1boy, national basketball association
10
  - text: 1girl, spy x
11
  - text: 1girl, absurdres
 
56
  nlp = pipeline('text-generation', model=model, tokenizer=tokenizer)
57
 
58
  # generate 10 samples using greedy search
59
+ outs = nlp(prompt, max_length=76, num_return_sequences=10, do_sample=True, repetition_penalty=1.2, temperature=0.7, top_k=4, early_stopping=True)
60
 
61
  print('\nInput:\n' + 100 * '-')
62
  print('\033[96m' + prompt + '\033[0m')