Merge pull request #7 from justjake/jake--use-split_einsum

Makefile: use split_einsum (fixes #6)
justjake · Jan 2, 2023 · 3043678 · 3043678
2 parents 7b34f1d + 6f4ee59
commit 3043678
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -2,7 +2,7 @@ PROJECT:=$(shell dirname $(realpath $(firstword $(MAKEFILE_LIST))))
 GIT_CLONE_SPARSE=GIT_LFS_SKIP_SMUDGE=1 git clone --depth 1
 
 # can be original or split_einsum
-ATTENTION_TYPE=original
+ATTENTION_TYPE=split_einsum
 ORIGINAL_MODEL_FILES=Unet.mlmodelc
 SPLIT_EINSUM_MODEL_FILES=UnetChunk1.mlmodelc UnetChunk2.mlmodelc