apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-02-15 20:50:12 -06:00
parent d3e8caade2
commit 841b18576c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert/diffusion_stable.py
View File

@ -126,9 +126,7 @@ def convert_diffusion_stable(
# UNET
if single_vae:
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "class_labels"]
unet_scale = torch.tensor(4).to(
device=ctx.training_device, dtype=torch.long
)
unet_scale = torch.tensor(4).to(device=ctx.training_device, dtype=torch.long)
else:
unet_inputs = ["sample", "timestep", "encoder_hidden_states", "return_dict"]
unet_scale = torch.tensor(False).to(