Fix linting
This commit is contained in:
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user