FredZhang7
commited on
Commit
·
b1182d2
1
Parent(s):
71aeaca
Update widget
Browse files
README.md
CHANGED
@@ -4,8 +4,8 @@ language:
|
|
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,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.
|
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')
|