mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 09:14:35 +02:00
update settings and fix references to state
This commit is contained in:
@@ -1,9 +1,9 @@
|
||||
import os
|
||||
|
||||
import torch
|
||||
from torch import nn
|
||||
from modules import devices, paths
|
||||
|
||||
|
||||
sd_vae_approx_model = None
|
||||
|
||||
|
||||
@@ -23,7 +23,7 @@ class VAEApprox(nn.Module):
|
||||
extra = 11
|
||||
try:
|
||||
x = nn.functional.interpolate(x, (x.shape[2] * 2, x.shape[3] * 2))
|
||||
x = nn.functional.pad(x, (extra, extra, extra, extra))
|
||||
x = nn.functional.pad(x, (extra, extra, extra, extra)) # pylint: disable=not-callable
|
||||
for layer in [self.conv1, self.conv2, self.conv3, self.conv4, self.conv5, self.conv6, self.conv7, self.conv8, ]:
|
||||
x = layer(x)
|
||||
x = nn.functional.leaky_relu(x, 0.1)
|
||||
@@ -34,7 +34,6 @@ class VAEApprox(nn.Module):
|
||||
|
||||
def model():
|
||||
global sd_vae_approx_model # pylint: disable=global-statement
|
||||
|
||||
if sd_vae_approx_model is None:
|
||||
from modules.shared import log
|
||||
model_path = os.path.join(paths.models_path, "VAE-approx", "model.pt")
|
||||
@@ -45,7 +44,6 @@ def model():
|
||||
sd_vae_approx_model.eval()
|
||||
sd_vae_approx_model.to(devices.device, devices.dtype)
|
||||
log.info(f"Loaded VAE-approx: model={model_path}")
|
||||
|
||||
return sd_vae_approx_model
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user