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()
|