Unverified Commit 742da319 authored by Sakura-Luna's avatar Sakura-Luna Committed by GitHub
Browse files

Minor changes

parent bd9b9d42

modules/sd_vae_taesd.py

+1 −1
Original line number Diff line number Diff line
@@ -71,6 +71,6 @@ def decode(): 
            sd_vae_taesd.eval()

            sd_vae_taesd.to(devices.device, devices.dtype)

        else:

            raise FileNotFoundError('Tiny AE mdoel not found')

            raise FileNotFoundError('Tiny AE model not found')



    return sd_vae_taesd.decoder

webui.py

+1 −1
Original line number Diff line number Diff line
@@ -151,7 +151,7 @@ def check_taesd(): 
    model_path = os.path.join(models_path, "VAE-approx", "taesd_decoder.pth")

    if not os.path.exists(model_path):

        print('download taesd model')

        torch.hub.download_url_to_file(model_url, os.path.dirname(model_path))

        torch.hub.download_url_to_file(model_url, model_path)





def initialize():

CRA Git | Maintained and supported by SUSTech CRA and CCSE