Update textual_inversio_with_blueloss.py
Browse files
textual_inversio_with_blueloss.py
CHANGED
@@ -11,7 +11,7 @@ from typing import List, Optional, Tuple, Union
|
|
11 |
from torchvision import transforms as tfms
|
12 |
from diffusers import StableDiffusionPipeline, AutoencoderKL, LMSDiscreteScheduler, UNet2DConditionModel
|
13 |
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
|
14 |
-
from focus_blur_utils import calculate_focus_blur_loss
|
15 |
# from transformers.modeling_attn_mask_utils import AttentionMaskConverter
|
16 |
|
17 |
class TextualInversion:
|
|
|
11 |
from torchvision import transforms as tfms
|
12 |
from diffusers import StableDiffusionPipeline, AutoencoderKL, LMSDiscreteScheduler, UNet2DConditionModel
|
13 |
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
|
14 |
+
# from focus_blur_utils import calculate_focus_blur_loss
|
15 |
# from transformers.modeling_attn_mask_utils import AttentionMaskConverter
|
16 |
|
17 |
class TextualInversion:
|