Vanisper commited on
Commit
eeff602
·
1 Parent(s): 8af6e20

feat: 确保text_encoder_projection_dim正确初始化

Browse files
Files changed (1) hide show
  1. utils.py +4 -0
utils.py CHANGED
@@ -306,6 +306,10 @@ def call(
306
  add_text_embeds = add_text_embeds.to(device)
307
  add_time_ids = add_time_ids.to(device).repeat(batch_size * num_images_per_prompt, 1)
308
 
 
 
 
 
309
  # 8. Denoising loop
310
  num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
311
 
 
306
  add_text_embeds = add_text_embeds.to(device)
307
  add_time_ids = add_time_ids.to(device).repeat(batch_size * num_images_per_prompt, 1)
308
 
309
+ # 确保 text_encoder_projection_dim 被正确初始化
310
+ if not hasattr(self, 'text_encoder_projection_dim'):
311
+ self.text_encoder_projection_dim = 128 # 或者其他适当的默认值
312
+
313
  # 8. Denoising loop
314
  num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
315