Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -140,6 +140,11 @@ class ModelWrapper:
|
|
140 |
"text_embeds": batch_pooled_prompt_embeds.squeeze(1)
|
141 |
}
|
142 |
|
|
|
|
|
|
|
|
|
|
|
143 |
eval_images = self.sample(noise=noise, unet_added_conditions=unet_added_conditions, prompt_embed=batch_prompt_embeds, fast_vae_decode=fast_vae_decode)
|
144 |
|
145 |
end_time = self._get_time()
|
|
|
140 |
"text_embeds": batch_pooled_prompt_embeds.squeeze(1)
|
141 |
}
|
142 |
|
143 |
+
|
144 |
+
print(noise.dtype)
|
145 |
+
print(batch_prompt_embeds.dtype)
|
146 |
+
|
147 |
+
|
148 |
eval_images = self.sample(noise=noise, unet_added_conditions=unet_added_conditions, prompt_embed=batch_prompt_embeds, fast_vae_decode=fast_vae_decode)
|
149 |
|
150 |
end_time = self._get_time()
|