Update app.py

app.py

@@ -219,6 +219,8 @@ class main():
                 print(self.text_encoder.device)
                 print(self.vae.device)
                 print(self.network.proj.device)
+                print(self.network.mean.device)
+                print(self.network.std.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