Commit 5dbd0355 authored by Kohaku-Blueleaf's avatar Kohaku-Blueleaf
Browse files

Fix linting

parent 64fd9163

modules/sd_samplers_common.py

+1 −1
Original line number Diff line number Diff line
@@ -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:

modules/sd_vae_consistency.py

+1 −3
Original line number Diff line number Diff line
@@ -5,8 +5,6 @@ Improved decoding for stable diffusion vaes. 
https://github.com/openai/consistencydecoder

"""

import os

import torch

import torch.nn as nn



from modules import devices, paths_internal, shared

from consistencydecoder import ConsistencyDecoder

CRA Git | Maintained and supported by SUSTech CRA and CCSE