wip

sci-ml/stable-diffusion-cpp/stable-diffusion-cpp-1.0.0.ebuild

@@ -13,16 +13,18 @@ EGIT_COMMIT="refs/tags/master-841-6b3edaa"
 LICENSE="MIT"
 SLOT="0"
 KEYWORDS="amd64 x86"
-IUSE="vulkan"
+IUSE="vulkan cuda"
 
 DEPEND="
-	sci-ml/ggml
 	media-libs/libwebm
 	vulkan? (
 		dev-util/glslang
 		dev-util/vulkan-headers
 		media-libs/vulkan-loader
 	)
+	cuda? (
+		dev-utl/nvidia-cuda-toolkit
+	)
 "
 
 RDEPEND="
@@ -31,11 +33,6 @@ RDEPEND="
 
 src_configure() {
 	local mycmakeargs=(
-		-DSD_BUILD_EXAMPLES=OFF
-		-DSD_BUILD_SHARED_LIBS=ON
-		-DSD_BUILD_SHARED_GGML_LIB=OFF
-		-DSD_USE_SYSTEM_GGML=ON
-
 		-DSD_WEBP=ON
 		-DSD_WEBM=ON
 	)
@@ -46,9 +43,17 @@ src_configure() {
 		)
 	fi
 
+	if use cuda; then
+		mycmakeargs+=(
+			-DSD_CUDA=ON
+		)
+	fi
+
 	cmake_src_configure
 }
 
 src_install() {
-	cmake_src_install
+	# cmake_src_install
+	dobin ${BUILD_DIR}/bin/sd-cli
+	dobin ${BUILD_DIR}/bin/sd-server
 }