Spaces:
Runtime error
Runtime error
code update
Browse files
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)
|