File size: 4,545 Bytes
1367e6b 3c7025e 1367e6b de87bdc e692727 a5515e4 7f1bd15 ecbcd62 7f1bd15 1367e6b 297482a 1367e6b e692727 1367e6b ecbcd62 1367e6b 3c7025e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 |
from dataclasses import dataclass
from typing import Dict, List
@dataclass
class Txt2TxtConfig:
default_system: str
default_model: Dict[str, int]
models: Dict[str, List[str]]
@dataclass
class Txt2ImgConfig:
default_model: Dict[str, int]
models: Dict[str, List[str]]
hidden_parameters: List[str]
negative_prompt: str
default_image_size: str
image_sizes: List[str]
default_aspect_ratio: str
aspect_ratios: List[str]
timeout: int = 60
@dataclass
class Config:
title: str
icon: str
layout: str
services: Dict[str, str]
txt2img: Txt2ImgConfig
txt2txt: Txt2TxtConfig
config = Config(
title="API Inference",
icon="⚡",
layout="wide",
services={
"Black Forest Labs": "https://api.bfl.ml/v1",
"Fal": "https://fal.run",
"Hugging Face": "https://api-inference.huggingface.co/models",
"Perplexity": "https://api.perplexity.ai",
"Together": "https://api.together.xyz/v1/images/generations",
},
txt2img=Txt2ImgConfig(
default_model={
"Black Forest Labs": 2,
"Fal": 0,
"Hugging Face": 2,
"Together": 0,
},
models={
# Model identifiers referenced in Text_to_Image.py
"Black Forest Labs": [
"flux-dev",
"flux-pro",
"flux-pro-1.1",
],
"Fal": [
"fal-ai/aura-flow",
"fal-ai/flux/dev",
"fal-ai/flux/schnell",
"fal-ai/flux-pro",
"fal-ai/flux-pro/v1.1",
"fal-ai/fooocus",
"fal-ai/kolors",
"fal-ai/stable-diffusion-v3-medium",
],
"Hugging Face": [
"black-forest-labs/flux.1-dev",
"black-forest-labs/flux.1-schnell",
"stabilityai/stable-diffusion-xl-base-1.0",
],
"Together": [
"black-forest-labs/FLUX.1-schnell-Free",
],
},
hidden_parameters=[
# sent to API but not shown in generation parameters accordion
"enable_safety_checker",
"max_sequence_length",
"num_images",
"output_format",
"performance",
"safety_tolerance",
"scheduler",
"sharpness",
"style",
"styles",
"sync_mode",
],
negative_prompt="ugly, unattractive, disfigured, deformed, mutated, malformed, blurry, grainy, noisy, oversaturated, undersaturated, overexposed, underexposed, worst quality, low details, lowres, watermark, signature, autograph, trademark, sloppy, cluttered",
default_image_size="square_hd",
image_sizes=[
"landscape_16_9",
"landscape_4_3",
"square_hd",
"portrait_4_3",
"portrait_16_9",
],
default_aspect_ratio="1024x1024",
aspect_ratios=[
"704x1408", # 1:2
"704x1344", # 11:21
"768x1344", # 4:7
"768x1280", # 3:5
"832x1216", # 13:19
"832x1152", # 13:18
"896x1152", # 7:9
"896x1088", # 14:17
"960x1088", # 15:17
"960x1024", # 15:16
"1024x1024",
"1024x960", # 16:15
"1088x960", # 17:15
"1088x896", # 17:14
"1152x896", # 9:7
"1152x832", # 18:13
"1216x832", # 19:13
"1280x768", # 5:3
"1344x768", # 7:4
"1344x704", # 21:11
"1408x704", # 2:1
],
),
txt2txt=Txt2TxtConfig(
default_system="You are a helpful assistant. Be precise and concise.",
default_model={
"Hugging Face": 4,
"Perplexity": 3,
},
models={
"Hugging Face": [
"codellama/codellama-34b-instruct-hf",
"meta-llama/llama-2-13b-chat-hf",
"meta-llama/meta-llama-3.1-405b-instruct-fp8",
"mistralai/mistral-7b-instruct-v0.2",
"nousresearch/nous-hermes-2-mixtral-8x7b-dpo",
],
"Perplexity": [
"llama-3.1-sonar-small-128k-chat",
"llama-3.1-sonar-large-128k-chat",
"llama-3.1-sonar-small-128k-online",
"llama-3.1-sonar-large-128k-online",
],
},
),
)
|