Added CLIPVisionModelWithProjection import
Browse fileshttps://github.com/huggingface/diffusers/pull/5993/files/69efda9fde2b874150ee3f2782e09009179605da
- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -23,7 +23,7 @@ from typing import Any, Callable, Dict, List, Optional, Union
|
|
| 23 |
|
| 24 |
import torch
|
| 25 |
import torch.nn.functional as F
|
| 26 |
-
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
|
| 27 |
|
| 28 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
| 29 |
from diffusers.models.attention import Attention, GatedSelfAttentionDense
|
|
|
|
| 23 |
|
| 24 |
import torch
|
| 25 |
import torch.nn.functional as F
|
| 26 |
+
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer, CLIPVisionModelWithProjection
|
| 27 |
|
| 28 |
from diffusers.models import AutoencoderKL, UNet2DConditionModel
|
| 29 |
from diffusers.models.attention import Attention, GatedSelfAttentionDense
|