use right tokenizer

mosaicml · Aug 27, 2024 · 7f97fd6 · 7f97fd6
1 parent e03fe0c
commit 7f97fd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffusion/callbacks/log_diffusion_images.py b/diffusion/callbacks/log_diffusion_images.py
@@ -108,7 +108,7 @@ def __init__(self,
 
                 tokenized_clip = clip_tokenizer(batch,
                                                 padding='max_length',
-                                                max_length=t5_tokenizer.model_max_length,
+                                                max_length=clip_tokenizer.model_max_length,
                                                 truncation=True,
                                                 return_tensors='pt')
                 clip_attention_mask = tokenized_clip['attention_mask'].cuda()