crumb
/

bloom-560m-RLHF-SD2-prompter

Text Generation

stable-diffusion

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

crumb commited on Dec 1, 2022

Commit

d330f54

•

1 Parent(s): 3056618

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ inference:
 Using RLHF (Reinforcement Learning from Human Feedback) to finetune [mrm8488/bloom-560m-finetuned-sd-prompts](https://hf.co/mrm8488/bloom-560m-finetuned-sd-prompts) further for SD2.0
 ```
-batch_size = 8
 learning_rate = 0.001
 ```

 Using RLHF (Reinforcement Learning from Human Feedback) to finetune [mrm8488/bloom-560m-finetuned-sd-prompts](https://hf.co/mrm8488/bloom-560m-finetuned-sd-prompts) further for SD2.0
 ```
+batch_size = 16
 learning_rate = 0.001
 ```