Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
|
@@ -14,8 +14,8 @@ from diffusers.utils.torch_utils import randn_tensor
|
|
| 14 |
from huggingface_hub import snapshot_download
|
| 15 |
from transformers import CLIPImageProcessor
|
| 16 |
|
| 17 |
-
from
|
| 18 |
-
from
|
| 19 |
from utils import (compute_vae_encodings, numpy_to_pil, prepare_image,
|
| 20 |
prepare_mask_image, resize_and_crop, resize_and_padding)
|
| 21 |
|
|
|
|
| 14 |
from huggingface_hub import snapshot_download
|
| 15 |
from transformers import CLIPImageProcessor
|
| 16 |
|
| 17 |
+
from attn_processor import SkipAttnProcessor
|
| 18 |
+
from utils import get_trainable_module, init_adapter
|
| 19 |
from utils import (compute_vae_encodings, numpy_to_pil, prepare_image,
|
| 20 |
prepare_mask_image, resize_and_crop, resize_and_padding)
|
| 21 |
|