Update app.py

app.py

@@ -39,7 +39,8 @@ scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedu
 # To the GPU we go!
 vae = vae.to(torch_device)
 text_encoder = text_encoder.to(torch_device)
-unet = unet.to(torch_device);
+unet = unet.to(torch_device)
+
 token_emb_layer = text_encoder.text_model.embeddings.token_embedding
 pos_emb_layer = text_encoder.text_model.embeddings.position_embedding
 position_ids = text_encoder.text_model.embeddings.position_ids[:, :77]
@@ -61,7 +62,7 @@ generator = torch.manual_seed(32)   # Seed generator to create the inital latent
 
 # Prep latents
 latents = torch.randn(
-  (batch_size, unet.in_channels, height // 8, width // 8),
+  (batch_size, unet.config.in_channels, height // 8, width // 8),
   generator=generator,
 )
 latents = latents.to(torch_device)