Commit ·
3724069
1
Parent(s): 8c4798e
Update README.md
Browse files
README.md
CHANGED
|
@@ -50,6 +50,7 @@ Check our jupyter notebooks with examples in `./examples` folder
|
|
| 50 |
|
| 51 |
```python
|
| 52 |
from diffusers import AutoPipelineForText2Image
|
|
|
|
| 53 |
|
| 54 |
pipe = AutoPipelineForText2Image.from_pretrained("kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16)
|
| 55 |
pipe.enable_model_cpu_offload()
|
|
@@ -65,6 +66,7 @@ image = pipe(prompt, num_inference_steps=25, generator=generator).images[0]
|
|
| 65 |
```python
|
| 66 |
from diffusers import AutoPipelineForImage2Image
|
| 67 |
from diffusers.utils import load_image
|
|
|
|
| 68 |
|
| 69 |
pipe = AutoPipelineForImage2Image.from_pretrained("kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16)
|
| 70 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 50 |
|
| 51 |
```python
|
| 52 |
from diffusers import AutoPipelineForText2Image
|
| 53 |
+
import torch
|
| 54 |
|
| 55 |
pipe = AutoPipelineForText2Image.from_pretrained("kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16)
|
| 56 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 66 |
```python
|
| 67 |
from diffusers import AutoPipelineForImage2Image
|
| 68 |
from diffusers.utils import load_image
|
| 69 |
+
import torch
|
| 70 |
|
| 71 |
pipe = AutoPipelineForImage2Image.from_pretrained("kandinsky-community/kandinsky-3", variant="fp16", torch_dtype=torch.float16)
|
| 72 |
pipe.enable_model_cpu_offload()
|