apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint, rename lookup table

This commit is contained in:
Sean Sube 2023-02-16 22:22:46 -06:00
parent f5ae9dd492
commit 388eb640c0
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/convert/diffusion_original.py
View File

@ -23,7 +23,6 @@ from logging import getLogger
from typing import Dict, List
import huggingface_hub.utils.tqdm
import safetensors.torch
import torch
from diffusers import (
AutoencoderKL,

4
api/onnx_web/diffusion/pipeline_onnx_stable_diffusion_upscale.py
View File

@ -38,7 +38,7 @@ ORT_TO_NP_TYPE = {
"tensor(double)": np.float64,
}
TORCH_DTYPES = {
ORT_TO_PT_TYPE = {
"float16": torch.float16,
"float32": torch.float32,
}
@ -112,7 +112,7 @@ class OnnxStableDiffusionUpscalePipeline(StableDiffusionUpscalePipeline):
prompt, device, num_images_per_prompt, do_classifier_free_guidance, negative_prompt
)
latents_dtype = TORCH_DTYPES[str(text_embeddings.dtype)]
latents_dtype = ORT_TO_PT_TYPE[str(text_embeddings.dtype)]
# 4. Preprocess image
image = preprocess(image)