Skip to content

Commit 523e63f

Browse files
Fix NoneType exception if vulkan tuning flags not detected. (huggingface#659)
(This goes on to produce compilation errors, but one step at a time)
1 parent 10630ab commit 523e63f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

web/models/stable_diffusion/cache_objects.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,10 @@
4848
)
4949
schedulers["SharkEulerDiscrete"].compile()
5050

51+
vulkan_triple_flags = get_vulkan_triple_flag()
52+
5153
# use tuned unet model in case of rdna3 cards.
52-
if "rdna3" in get_vulkan_triple_flag():
54+
if vulkan_triple_flags and "rdna3" in vulkan_triple_flags:
5355
args.use_tuned = True
5456

5557
# set iree-runtime flags

0 commit comments

Comments
 (0)