pszemraj
/

distilgpt2-magicprompt-SD

Text Generation

Generated from Trainer

stable diffusion

prompt engineering

text-generation-inference

Model card Files Files and versions

pszemraj commited on Nov 9, 2022

Commit

37a1a79

·

1 Parent(s): 770d173

Update README.md

Files changed (1) hide show

README.md +4 -3

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ widget:
    example_title: "creative prompt"
 parameters:
   min_length: 16
-  max_length: 96
   no_repeat_ngram_size: 1
   do_sample: True
 ---
@@ -48,7 +48,7 @@ Results in (_DALL-E, but you get the idea_):
 ![example](https://i.ibb.co/WkmTnZD/image.png)
 - this `distilgpt2` version is probably small/fast enough to be used locally on CPU!
 ## basic usage
@@ -73,7 +73,8 @@ generator = pipeline(
 prompt = "The Answer to Why"
 result = generator(
     prompt,
-)  # generate
 print(result[0]["generated_text"])
 ```

    example_title: "creative prompt"
 parameters:
   min_length: 16
+  max_new_tokens: 24
   no_repeat_ngram_size: 1
   do_sample: True
 ---
 ![example](https://i.ibb.co/WkmTnZD/image.png)
+<br><br>
 - this `distilgpt2` version is probably small/fast enough to be used locally on CPU!
 ## basic usage
 prompt = "The Answer to Why"
 result = generator(
     prompt,
+    max_new_tokens=24,
+)  # generate, adjust/add kwargs as needed
 print(result[0]["generated_text"])
 ```