github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix check for use_tuned conditions (#1252)

Browse Source
This commit is contained in:
yzhang93
2023-03-27 11:21:25 -07:00
committed by GitHub
parent 7619e76bbd
commit 8f84258fb8
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
apps/stable_diffusion/src/utils/utils.py
View File

@@ -270,7 +270,8 @@ def set_init_device_flags():
if (
args.precision != "fp16"
or args.height not in [512, 768]
or args.width not in [512, 768]
or (args.height == 512 and args.width != 512)
or (args.height == 768 and args.width != 768)
or args.batch_size != 1
or ("vulkan" not in args.device and "cuda" not in args.device)
):