Shaleen123
commited on
Commit
•
bb4477a
1
Parent(s):
aef55fe
Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ The Kandinsky model is created by [Arseniy Shakhmatov](https://github.com/cene55
|
|
21 |
Kandinsky 2.2 is available in diffusers!
|
22 |
|
23 |
```python
|
24 |
-
pip install diffusers
|
25 |
```
|
26 |
### Text to image
|
27 |
|
@@ -29,14 +29,13 @@ pip install diffusers transformers accelerate
|
|
29 |
from diffusers import AutoPipelineForText2Image
|
30 |
import torch
|
31 |
|
32 |
-
pipe = AutoPipelineForText2Image.from_pretrained("
|
33 |
pipe = pipe.to("cuda")
|
34 |
|
35 |
-
prompt = "
|
36 |
-
negative_prompt = "low quality, bad quality"
|
37 |
-
|
38 |
-
|
39 |
-
image.save("portrait.png")
|
40 |
```
|
41 |
|
42 |
![img](https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/kandinskyv22/%20blue%20eyes.png)
|
|
|
21 |
Kandinsky 2.2 is available in diffusers!
|
22 |
|
23 |
```python
|
24 |
+
pip install diffusers bitsandbytes
|
25 |
```
|
26 |
### Text to image
|
27 |
|
|
|
29 |
from diffusers import AutoPipelineForText2Image
|
30 |
import torch
|
31 |
|
32 |
+
pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16)
|
33 |
pipe = pipe.to("cuda")
|
34 |
|
35 |
+
prompt = input("Enter your prompt - ")
|
36 |
+
negative_prompt = "low quality, bad quality, boring, ugly"
|
37 |
+
image_sub = pipe(prompt=prompt, negative_prompt=negative_prompt, prior_guidance_scale =1.25, width=1024, height=1024, num_inference_steps = 75).images[0]
|
38 |
+
image_sub
|
|
|
39 |
```
|
40 |
|
41 |
![img](https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/kandinskyv22/%20blue%20eyes.png)
|