Spaces:

Lifeinhockey
/

Diffusion_Models

Running

Lifeinhockey commited on Feb 28

Commit

3f4aa4c

verified ·

1 Parent(s): 8edb646

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -566,7 +566,7 @@ def infer(
                         print('use_Tiny_VAE = ', use_Tiny_VAE)
-                        f Tiny_VAE == "sd-vae-ft-mse":
                             VAE_id = "stabilityai/sd-vae-ft-mse"
                         generator = torch.Generator(device).manual_seed(seed)

                         print('use_Tiny_VAE = ', use_Tiny_VAE)
+                        if Tiny_VAE == "sd-vae-ft-mse":
                             VAE_id = "stabilityai/sd-vae-ft-mse"
                         generator = torch.Generator(device).manual_seed(seed)