Update README.md
Browse files
README.md
CHANGED
|
@@ -81,7 +81,7 @@ pip install transformers==4.57.1 torchvision==0.21.0 torch==2.6.0 diffusers==0.3
|
|
| 81 |
|
| 82 |
# Quick start
|
| 83 |
|
| 84 |
-
**Note:** When using this distilled model,
|
| 85 |
|
| 86 |
```python
|
| 87 |
from PIL import Image
|
|
@@ -102,6 +102,8 @@ model_path = snapshot_download(
|
|
| 102 |
editor = ImageEditor(
|
| 103 |
checkpoint_path=model_path,
|
| 104 |
num_inference_steps=20,
|
|
|
|
|
|
|
| 105 |
device="cuda:0",
|
| 106 |
)
|
| 107 |
|
|
|
|
| 81 |
|
| 82 |
# Quick start
|
| 83 |
|
| 84 |
+
**Note:** When using this distilled model, please set `image_guidance_scale` and `guidance_scale` to 0.0 to disable CFG.
|
| 85 |
|
| 86 |
```python
|
| 87 |
from PIL import Image
|
|
|
|
| 102 |
editor = ImageEditor(
|
| 103 |
checkpoint_path=model_path,
|
| 104 |
num_inference_steps=20,
|
| 105 |
+
image_guidance_scale=0.0,
|
| 106 |
+
guidance_scale=0.0,
|
| 107 |
device="cuda:0",
|
| 108 |
)
|
| 109 |
|