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 # SINGLE VAE
vae_only = pipeline.vae vae_only = pipeline.vae
vae_latent_channels = vae_only.config.latent_channels vae_latent_channels = vae_only.config.latent_channels
vae_out_channels = vae_only.config.out_channels
# forward only through the decoder part # forward only through the decoder part
vae_only.forward = vae_only.decode vae_only.forward = vae_only.decode
onnx_export( onnx_export(
@ -266,7 +265,6 @@ def convert_diffusion_diffusers(
# VAE DECODER # VAE DECODER
vae_decoder = pipeline.vae vae_decoder = pipeline.vae
vae_latent_channels = vae_decoder.config.latent_channels vae_latent_channels = vae_decoder.config.latent_channels
vae_out_channels = vae_decoder.config.out_channels
# forward only through the decoder part # forward only through the decoder part
vae_decoder.forward = vae_encoder.decode vae_decoder.forward = vae_encoder.decode
onnx_export( 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 from typing import Any, Optional, Tuple
import numpy as np import numpy as np
import torch
from diffusers import ( from diffusers import (
DDIMScheduler, DDIMScheduler,
DDPMScheduler, DDPMScheduler,