Update app.py

app.py

@@ -219,8 +219,11 @@ class main():
                 print(self.text_encoder.device)
                 print(self.vae.device)
                 print(self.network.proj.device)
+                print(self.network.v.device)
+                print(text_embeddings.device)
                 noise_pred = self.unet(latent_model_input, t, encoder_hidden_states=text_embeddings, timestep_cond= None).sample
-                #guidance
+            print("after inference")
+            #guidance
             noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
             noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
             latents = noise_scheduler.step(noise_pred, t, latents).prev_sample