Update syngen_diffusion_pipeline.py

syngen_diffusion_pipeline.py

@@ -19,7 +19,7 @@ from diffusers.utils import (
 )
 from transformers import CLIPTextModel, CLIPTokenizer, CLIPFeatureExtractor
 
-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
+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, extract_attribution_indices_with_verbs, extract_attribution_indices_with_verb_root
 
 
 logger = logging.get_logger(__name__)