pszemraj
/

gpt-neo-125M-magicprompt-SD

Text Generation

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Community

pszemraj commited on Nov 9, 2022

Commit

a6b830e

·

1 Parent(s): 5f2109f

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -21,12 +21,12 @@ parameters:
   max_length: 96
   no_repeat_ngram_size: 1
   do_sample: True
-  temperature: 0.9
   top_k: 60
   top_p: 0.95
-  repetition_penalty: 3.5
-  length_penalty: 0.9
 ---
 # neo-125M-promptgen-v1-act
 This model is a fine-tuned version of [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) on the Gustavosta/Stable-Diffusion-Prompts dataset.

   max_length: 96
   no_repeat_ngram_size: 1
   do_sample: True
+  temperature: 1.1
   top_k: 60
   top_p: 0.95
+  repetition_penalty: 7.3
 ---
 # neo-125M-promptgen-v1-act
 This model is a fine-tuned version of [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) on the Gustavosta/Stable-Diffusion-Prompts dataset.