Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -213,6 +213,7 @@ class main():
|
|
213 |
for i,t in enumerate(tqdm.tqdm(self.noise_scheduler.timesteps)):
|
214 |
latent_model_input = torch.cat([latents] * 2)
|
215 |
latent_model_input = self.noise_scheduler.scale_model_input(latent_model_input, timestep=t)
|
|
|
216 |
with self.network:
|
217 |
noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
|
218 |
#guidance
|
|
|
213 |
for i,t in enumerate(tqdm.tqdm(self.noise_scheduler.timesteps)):
|
214 |
latent_model_input = torch.cat([latents] * 2)
|
215 |
latent_model_input = self.noise_scheduler.scale_model_input(latent_model_input, timestep=t)
|
216 |
+
print("going into inference 1")
|
217 |
with self.network:
|
218 |
noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
|
219 |
#guidance
|