Spaces:
Running
on
Zero
Running
on
Zero
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 |
|