2024-03-20 11:44:15 +08:00
|
|
|
import os
|
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
from pydantic import BaseSettings
|
|
|
|
|
|
|
|
|
|
BASE_DIR = os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))
|
|
|
|
|
load_dotenv(os.path.join(BASE_DIR, '.env'))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
|
|
|
PROJECT_NAME = os.getenv('PROJECT_NAME', 'FASTAPI BASE')
|
|
|
|
|
SECRET_KEY = os.getenv('SECRET_KEY', '')
|
|
|
|
|
API_PREFIX = ''
|
|
|
|
|
BACKEND_CORS_ORIGINS = ['*']
|
|
|
|
|
DATABASE_URL = os.getenv('SQL_DATABASE_URL', '')
|
|
|
|
|
ACCESS_TOKEN_EXPIRE_SECONDS: int = 60 * 60 * 24 * 7 # Token expired after 7 days
|
|
|
|
|
SECURITY_ALGORITHM = 'HS256'
|
|
|
|
|
LOGGING_CONFIG_FILE = os.path.join(BASE_DIR, 'logging_env.py')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
settings = Settings()
|
|
|
|
|
|
|
|
|
|
ckpt = 'service/super_resolution_ccsr/weights/real-world_ccsr.ckpt'
|
|
|
|
|
config = 'service/super_resolution_ccsr/configs/model/ccsr_stage2.yaml'
|
|
|
|
|
steps = 45
|
|
|
|
|
sr_scale = 4
|
|
|
|
|
repeat_times = 1
|
|
|
|
|
tiled = False
|
|
|
|
|
tile_size = 512
|
|
|
|
|
tile_stride = 256
|
|
|
|
|
color_fix_type = "adain"
|
|
|
|
|
t_max = 0.6667
|
|
|
|
|
t_min = 0.3333
|
|
|
|
|
show_lq = False
|
|
|
|
|
skip_if_exist = False
|
|
|
|
|
seed = 233
|
|
|
|
|
device = "cuda"
|
|
|
|
|
tile_diffusion = False #
|
|
|
|
|
tile_diffusion_size = 512
|
|
|
|
|
tile_diffusion_stride = 256
|
|
|
|
|
tile_vae = True
|
|
|
|
|
vae_decoder_tile_size = 224
|
|
|
|
|
vae_encoder_tile_size = 1024
|
|
|
|
|
strength = 1
|
|
|
|
|
# minio 配置
|
|
|
|
|
sr_bucket = "test"
|
|
|
|
|
MINIO_IP = "www.minio.aida.com.hk"
|
|
|
|
|
MINIO_PORT = 9000
|
|
|
|
|
MINIO_ACCESS = 'vXKFLSJkYeEq2DrSZvkB'
|
|
|
|
|
MINIO_SECRET = 'uKTZT3x7C43WvPN9QTc99DiRkwddWZrG9Uh3JVlR'
|
|
|
|
|
MINIO_SECURE = True
|
|
|
|
|
# input = 'preprocess_img/input_x2' # 这个值需要被函数参数覆盖
|
|
|
|
|
# output = '/path/to/output' # 这个值将被函数参数覆盖
|
2024-03-20 12:04:25 +08:00
|
|
|
LOGS_PATH = "app/logs/errors.log"
|
