Spaces:
Sleeping
Sleeping
Update syngen_diffusion_pipeline.py
Browse files
syngen_diffusion_pipeline.py
CHANGED
@@ -19,6 +19,8 @@ from diffusers.utils import (
|
|
19 |
)
|
20 |
from transformers import CLIPTextModel, CLIPTokenizer, CLIPFeatureExtractor
|
21 |
|
|
|
|
|
22 |
|
23 |
logger = logging.get_logger(__name__)
|
24 |
|
|
|
19 |
)
|
20 |
from transformers import CLIPTextModel, CLIPTokenizer, CLIPFeatureExtractor
|
21 |
|
22 |
+
from compute_loss import get_attention_map_index_to_wordpiece, split_indices, calculate_positive_loss, calculate_negative_loss, get_indices, start_token, end_token, align_wordpieces_indices, extract_attribution_indices
|
23 |
+
|
24 |
|
25 |
logger = logging.get_logger(__name__)
|
26 |
|