OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-07 06:02:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix linting

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_common.py
View File

@ -55,7 +55,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None):
with devices.autocast(), torch.no_grad(): with devices.autocast(), torch.no_grad():
x_sample = sd_vae_consistency.decoder_model()( x_sample = sd_vae_consistency.decoder_model()(
sample.to(devices.device, devices.dtype)/0.18215, sample.to(devices.device, devices.dtype)/0.18215,
schedule=[1.0] schedule=[1.0],
) )
sd_vae_consistency.unload() sd_vae_consistency.unload()
else: else:

2
modules/sd_vae_consistency.py
View File

@ -5,8 +5,6 @@ Improved decoding for stable diffusion vaes.
https://github.com/openai/consistencydecoder https://github.com/openai/consistencydecoder
""" """
import os import os
import torch
import torch.nn as nn
from modules import devices, paths_internal, shared from modules import devices, paths_internal, shared
from consistencydecoder import ConsistencyDecoder from consistencydecoder import ConsistencyDecoder