github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

Remove fluxdev from AI Generation block due to licence restrictions. (#8222)

Browse Source
This commit is contained in:
Toran Bruce Richards
2024-09-30 10:54:40 +01:00
committed by GitHub
parent 22fb9bc635
commit 9dc7bb8497
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
autogpt_platform/backend/backend/blocks/replicate_flux_advanced.py
View File

@@ -11,14 +11,12 @@ from backend.data.model import BlockSecret, SchemaField, SecretField
class ReplicateFluxModelName(str, Enum):
FLUX_SCHNELL = ("Flux Schnell",)
FLUX_PRO = ("Flux Pro",)
FLUX_DEV = ("Flux Dev",)
@property
def api_name(self):
api_names = {
ReplicateFluxModelName.FLUX_SCHNELL: "black-forest-labs/flux-schnell",
ReplicateFluxModelName.FLUX_PRO: "black-forest-labs/flux-pro",
ReplicateFluxModelName.FLUX_DEV: "black-forest-labs/flux-dev",
}
return api_names[self]