Fix linting

This commit is contained in:
Kohaku-Blueleaf
2023-11-07 11:00:24 +08:00
parent 64fd916334
commit 5dbd0355b0
2 changed files with 2 additions and 4 deletions
+1 -3
View File
@@ -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
@@ -32,4 +30,4 @@ def decoder_model():
def unload():
global sd_vae_consistency_models
if sd_vae_consistency_models is not None:
sd_vae_consistency_models.ckpt.to('cpu')
sd_vae_consistency_models.ckpt.to('cpu')