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)
|