added sdxl example for replicate sdxl

This commit is contained in:
Believethehype
2023-12-06 13:46:19 +01:00
parent 90865e6cdb
commit a9ffd4299c
7 changed files with 182 additions and 5 deletions

View File

@@ -15,6 +15,7 @@ import tasks.textextraction_pdf as textextraction_pdf
import tasks.textextraction_google as textextraction_google
import tasks.translation_google as translation_google
import tasks.translation_libretranslate as translation_libretranslate
from tasks import imagegeneration_replicate
from utils.admin_utils import AdminConfig
from utils.backend_utils import keep_alive
@@ -75,6 +76,12 @@ def playground():
bot_config.SUPPORTED_DVMS.append(dalle)
dalle.run()
if os.getenv("REPLICATE_API_TOKEN") is not None and os.getenv("REPLICATE_API_TOKEN") != "":
sdxlreplicate = imagegeneration_replicate.build_example("Stable Diffusion XL", "replicate_sdxl", admin_config)
bot_config.SUPPORTED_DVMS.append(sdxlreplicate)
sdxlreplicate.run()
#Let's define a function so we can add external DVMs to our bot, we will instanciate it afterwards