amildravid4292 commited on
Commit
6bbb81f
·
verified ·
1 Parent(s): 8601f5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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