Commit
•
73eca23
1
Parent(s):
0250758
Update app.py
Browse files
app.py
CHANGED
@@ -46,11 +46,13 @@ def main(
|
|
46 |
|
47 |
description = \
|
48 |
"""
|
|
|
|
|
49 |
Generate variations on an input image using a fine-tuned version of Stable Diffision.
|
50 |
Trained by [Justin Pinkney](https://www.justinpinkney.com) ([@Buntworthy](https://twitter.com/Buntworthy)) at [Lambda](https://lambdalabs.com/)
|
51 |
|
52 |
This version has been ported to 🤗 Diffusers library, see more details on how to use this version in the [Lambda Diffusers repo](https://github.com/LambdaLabsML/lambda-diffusers).
|
53 |
-
|
54 |
|
55 |
![](https://raw.githubusercontent.com/justinpinkney/stable-diffusion/main/assets/im-vars-thin.jpg)
|
56 |
|
@@ -66,8 +68,8 @@ This creates images which have the same rough style and content, but different d
|
|
66 |
This is a totally different approach to the img2img script of the original Stable Diffusion and gives very different results.
|
67 |
|
68 |
The model was fine tuned on the [LAION aethetics v2 6+ dataset](https://laion.ai/blog/laion-aesthetics/) to accept the new conditioning.
|
69 |
-
Training was done on
|
70 |
-
More details on the
|
71 |
"""
|
72 |
|
73 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
46 |
|
47 |
description = \
|
48 |
"""
|
49 |
+
__Now using Image Variations v2!__
|
50 |
+
|
51 |
Generate variations on an input image using a fine-tuned version of Stable Diffision.
|
52 |
Trained by [Justin Pinkney](https://www.justinpinkney.com) ([@Buntworthy](https://twitter.com/Buntworthy)) at [Lambda](https://lambdalabs.com/)
|
53 |
|
54 |
This version has been ported to 🤗 Diffusers library, see more details on how to use this version in the [Lambda Diffusers repo](https://github.com/LambdaLabsML/lambda-diffusers).
|
55 |
+
For the original training code see [this repo](https://github.com/justinpinkney/stable-diffusion).
|
56 |
|
57 |
![](https://raw.githubusercontent.com/justinpinkney/stable-diffusion/main/assets/im-vars-thin.jpg)
|
58 |
|
|
|
68 |
This is a totally different approach to the img2img script of the original Stable Diffusion and gives very different results.
|
69 |
|
70 |
The model was fine tuned on the [LAION aethetics v2 6+ dataset](https://laion.ai/blog/laion-aesthetics/) to accept the new conditioning.
|
71 |
+
Training was done on 8xA100 GPUs on [Lambda GPU Cloud](https://lambdalabs.com/service/gpu-cloud).
|
72 |
+
More details are on the [model card](https://huggingface.co/lambdalabs/sd-image-variations-diffusers).
|
73 |
"""
|
74 |
|
75 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|