Merge pull request #10 from Manifest-Holdings/adding_optional_flag_for_together_model

adding_optional_flag_for_together_model

Author: Cooops

.env.example

@@ -351,10 +351,11 @@ MEDIUM_LMSTUDIO_MODEL= # Default: hermes-3-llama-3.1-8b
 LARGE_LMSTUDIO_MODEL=  # Default: hermes-3-llama-3.1-8b
 
 # Remaining Provider Configurations
-GOOGLE_GENERATIVE_AI_API_KEY= # Gemini API key
-ALI_BAILIAN_API_KEY=          # Ali Bailian API Key
-NANOGPT_API_KEY=              # NanoGPT API Key
-TOGETHER_API_KEY=             # Together API Key
+GOOGLE_GENERATIVE_AI_API_KEY=   # Gemini API key
+ALI_BAILIAN_API_KEY=            # Ali Bailian API Key
+NANOGPT_API_KEY=                # NanoGPT API Key
+TOGETHER_API_KEY=               # Together API Key
+TOGETHER_IMAGE_MODEL=           # Default: black-forest-labs/FLUX.1-schnell
 
 ######################################
 #### Crypto Plugin Configurations ####

packages/core/src/models.ts

@@ -295,7 +295,7 @@ export const models: Models = {
                 name: "togethercomputer/m2-bert-80M-32k-retrieval",
             },
             [ModelClass.IMAGE]: {
-                name: "black-forest-labs/FLUX.1-schnell",
+                name: settings.TOGETHER_IMAGE_MODEL || "black-forest-labs/FLUX.1-schnell",
                 steps: 4,
             },
         },