Shaleen123 commited on
Commit
bb4477a
1 Parent(s): aef55fe

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -7
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 transformers accelerate
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("kandinsky-community/kandinsky-2-2-decoder", torch_dtype=torch.float16)
33
  pipe = pipe.to("cuda")
34
 
35
- prompt = "portrait of a young women, blue eyes, cinematic"
36
- negative_prompt = "low quality, bad quality"
37
-
38
- image = pipe(prompt=prompt, negative_prompt=negative_prompt, prior_guidance_scale =1.0, height=768, width=768).images[0]
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)