mkthoma commited on
Commit
9acff6b
·
1 Parent(s): 1782f93

code update

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -38,6 +38,13 @@ vae = vae.to(torch_device)
38
  text_encoder = text_encoder.to(torch_device)
39
  unet = unet.to(torch_device)
40
 
 
 
 
 
 
 
 
41
  # Prep Scheduler
42
  def set_timesteps(scheduler, num_inference_steps):
43
  scheduler.set_timesteps(num_inference_steps)
 
38
  text_encoder = text_encoder.to(torch_device)
39
  unet = unet.to(torch_device)
40
 
41
+ seed_values = [8,16,50,80,128]
42
+ height = 512 # default height of Stable Diffusion
43
+ width = 512 # default width of Stable Diffusion
44
+ num_inference_steps = 1 # Number of denoising steps
45
+ guidance_scale = 7.5 # Scale for classifier-free guidance
46
+ num_styles = len(style_files)
47
+
48
  # Prep Scheduler
49
  def set_timesteps(scheduler, num_inference_steps):
50
  scheduler.set_timesteps(num_inference_steps)