Update app.py
Browse files
app.py
CHANGED
|
@@ -220,6 +220,7 @@ class main():
|
|
| 220 |
for i,t in enumerate(tqdm.tqdm(self.noise_scheduler.timesteps)):
|
| 221 |
latent_model_input = torch.cat([latents] * 2)
|
| 222 |
latent_model_input = self.noise_scheduler.scale_model_input(latent_model_input, timestep=t)
|
|
|
|
| 223 |
with network:
|
| 224 |
noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
|
| 225 |
|
|
|
|
| 220 |
for i,t in enumerate(tqdm.tqdm(self.noise_scheduler.timesteps)):
|
| 221 |
latent_model_input = torch.cat([latents] * 2)
|
| 222 |
latent_model_input = self.noise_scheduler.scale_model_input(latent_model_input, timestep=t)
|
| 223 |
+
print(text_embeddings.dtype)
|
| 224 |
with network:
|
| 225 |
noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
|
| 226 |
|