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
|