From 3ee6d122eb13f4c81e462f2502eb78ef7a2aa820 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Mon, 4 Mar 2024 18:46:06 +0100 Subject: [PATCH] Add pillow --- backend/python/diffusers/install.sh | 3 ++- core/backend/image.go | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/backend/python/diffusers/install.sh b/backend/python/diffusers/install.sh index 4f095912a131..c8e8b4518b01 100755 --- 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 index f7209f9d2a7c..478c32fdb6d6 100644 --- 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),