Spaces:
Sleeping
Sleeping
feat: 确保text_encoder_projection_dim正确初始化
Browse files
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 |
|