apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint, remove unused

This commit is contained in:
Sean Sube 2023-02-28 23:05:17 -06:00
parent 2f4ab20f61
commit 3f9f94fcb5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/convert/diffusion/diffusers.py
View File

@ -216,7 +216,6 @@ def convert_diffusion_diffusers(
# SINGLE VAE
vae_only = pipeline.vae
vae_latent_channels = vae_only.config.latent_channels
vae_out_channels = vae_only.config.out_channels
# forward only through the decoder part
vae_only.forward = vae_only.decode
onnx_export(
@ -266,7 +265,6 @@ def convert_diffusion_diffusers(
# VAE DECODER
vae_decoder = pipeline.vae
vae_latent_channels = vae_decoder.config.latent_channels
vae_out_channels = vae_decoder.config.out_channels
# forward only through the decoder part
vae_decoder.forward = vae_encoder.decode
onnx_export(

1
api/onnx_web/diffusion/load.py
View File

@ -3,7 +3,6 @@ from os import path
from typing import Any, Optional, Tuple
import numpy as np
import torch
from diffusers import (
DDIMScheduler,
DDPMScheduler,