Spaces:

sunjuice
/

FashionFlow

Sleeping

sunjuice commited on Nov 6, 2024

Commit

5ff190a

1 Parent(s): 95ba07b

fixed gradio

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ def forward_diffusion_sample(x_0, t):
     return sqrt_alphas_cumprod_t.to(device) * x_0.to(device) \
     + sqrt_one_minus_alphas_cumprod_t.to(device) * noise.to(device), noise.to(device)
-T = 10
 betas = cosine_beta_schedule(timesteps=T)
 # Pre-calculate different terms for closed form
 alphas = 1. - betas

     return sqrt_alphas_cumprod_t.to(device) * x_0.to(device) \
     + sqrt_one_minus_alphas_cumprod_t.to(device) * noise.to(device), noise.to(device)
+T = 1000
 betas = cosine_beta_schedule(timesteps=T)
 # Pre-calculate different terms for closed form
 alphas = 1. - betas