Add pillow

mudler · Mar 4, 2024 · 3ee6d12 · 3ee6d12
1 parent b4f1d46
commit 3ee6d12
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/backend/python/diffusers/install.sh b/backend/python/diffusers/install.sh
@@ -24,7 +24,8 @@ if [ -d "/opt/intel" ]; then
                 diffusers==0.24.0 \
                 transformers>=4.25.1 \
                 accelerate \
-                compel==2.0.2
+                compel==2.0.2 \
+                Pillow
 fi
 
 if [ "$PIP_CACHE_PURGE" = true ] ; then

diff --git a/core/backend/image.go b/core/backend/image.go
@@ -13,7 +13,7 @@ func ImageGeneration(height, width, mode, step, seed int, positive_prompt, negat
 	opts := modelOpts(backendConfig, appConfig, []model.Option{
 		model.WithBackendString(backendConfig.Backend),
 		model.WithAssetDir(appConfig.AssetsDestination),
-		model.WithThreads(uint32(appConfig.Threads)),
+		model.WithThreads(uint32(backendConfig.Threads)),
 		model.WithContext(appConfig.Context),
 		model.WithModel(backendConfig.Model),
 		model.WithLoadGRPCLoadModelOpts(gRPCOpts),