AYYasaswini commited on
Commit
8335dcb
1 Parent(s): e6d8c8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -52,6 +52,12 @@ def set_timesteps(scheduler, num_inference_steps):
52
  scheduler.timesteps = scheduler.timesteps.to(torch.float32) # minor fix to ensure MPS compatibility, fixed in diffusers PR 3925
53
 
54
  batch_size = 1
 
 
 
 
 
 
55
 
56
  # Prep latents
57
  latents = torch.randn(
 
52
  scheduler.timesteps = scheduler.timesteps.to(torch.float32) # minor fix to ensure MPS compatibility, fixed in diffusers PR 3925
53
 
54
  batch_size = 1
55
+ height = 512 # default height of Stable Diffusion
56
+ width = 512 # default width of Stable Diffusion
57
+ num_inference_steps = 10 # Number of denoising steps
58
+ guidance_scale = 7.5 # Scale for classifier-free guidance
59
+ generator = torch.manual_seed(32) # Seed generator to create the inital latent noise
60
+
61
 
62
  # Prep latents
63
  latents = torch.randn(