Fix linting

2023-11-07 11:00:24 +08:00
parent 64fd916334
commit 5dbd0355b0
2 changed files with 2 additions and 4 deletions
@@ -55,7 +55,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None):
        with devices.autocast(), torch.no_grad():
            x_sample = sd_vae_consistency.decoder_model()(
                sample.to(devices.device, devices.dtype)/0.18215,
-                schedule=[1.0]
+                schedule=[1.0],
            )
        sd_vae_consistency.unload()
    else: