justinpinkney commited on
Commit
c46d782
1 Parent(s): c337526

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -14
README.md CHANGED
@@ -68,7 +68,7 @@ This model is fine tuned from Stable Diffusion v1-3 where the text encoder has b
68
  - **Batch:** 6 x 4 = 24
69
  - **Learning rate:** warmup to 0.0001 for 1,000 steps and then kept constant
70
 
71
- Training was done using a [modified version of the original Stable Diffusion training code]((https://github.com/justinpinkney/stable-diffusion)
72
 
73
 
74
  # Uses
@@ -140,17 +140,4 @@ Specifically, the checker compares the class probability of harmful concepts in
140
  The concepts are passed into the model with the generated image and compared to a hand-engineered weight for each NSFW concept.
141
 
142
 
143
- ## Citation
144
-
145
- ```bibtex
146
- @InProceedings{Rombach_2022_CVPR,
147
- author = {Rombach, Robin and Blattmann, Andreas and Lorenz, Dominik and Esser, Patrick and Ommer, Bj\"orn},
148
- title = {High-Resolution Image Synthesis With Latent Diffusion Models},
149
- booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
150
- month = {June},
151
- year = {2022},
152
- pages = {10684-10695}
153
- }
154
- ```
155
-
156
  *This model card was written by: Justin Pinkney and is based on the [Stable Diffusion model card](https://huggingface.co/CompVis/stable-diffusion-v1-4).*
 
68
  - **Batch:** 6 x 4 = 24
69
  - **Learning rate:** warmup to 0.0001 for 1,000 steps and then kept constant
70
 
71
+ Training was done using a [modified version of the original Stable Diffusion training code]((https://github.com/justinpinkney/stable-diffusion), the original version of the weights is [here](https://huggingface.co/lambdalabs/stable-diffusion-image-conditioned).
72
 
73
 
74
  # Uses
 
140
  The concepts are passed into the model with the generated image and compared to a hand-engineered weight for each NSFW concept.
141
 
142
 
 
 
 
 
 
 
 
 
 
 
 
 
 
143
  *This model card was written by: Justin Pinkney and is based on the [Stable Diffusion model card](https://huggingface.co/CompVis/stable-diffusion-v1-4).*