Spaces:
Sleeping
Sleeping
File size: 1,063 Bytes
9bc957e 7a1d414 0106bb4 7a1d414 |
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 |
"""Application runtime settings for sdgen.
AppSettings reads configuration values from environment variables at
process start and exposes them as strongly typed attributes.
"""
from __future__ import annotations
import os
from dataclasses import dataclass
@dataclass
class AppSettings:
"""Config values for the Stable Diffusion app.
Supported environment variables:
- MODEL_ID: HuggingFace model name
- XFORMERS: 1/0 to enable xformers
- WARMUP: 1/0 to warm up CUDA kernels
- PORT: server port for Gradio
- HOST: server host address
- SHARE: enable Gradio public sharing link
"""
model_id1: str = os.getenv("MODEL_ID1", "runwayml/stable-diffusion-v1-5")
model_id2: str = os.getenv("MODEL_ID2", "stabilityai/stable-diffusion-turbo")
enable_xformers: bool = bool(int(os.getenv("XFORMERS", "0")))
warmup: bool = bool(int(os.getenv("WARMUP", "1")))
server_port: int = int(os.getenv("PORT", "7860"))
server_host: str = os.getenv("HOST", "0.0.0.0")
share: bool = bool(int(os.getenv("SHARE", "1")))
|