File size: 2,289 Bytes
aaa0b94 |
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 |
import os
import numpy as np
HF_TOKEN = os.getenv('HF_TOKEN')
UPLOAD_REPO_ID = os.getenv('UPLOAD_REPO_ID')
UPLOAD_RESULT_IMAGE = os.getenv('UPLOAD_RESULT_IMAGE') == '1'
# UI options
SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON', '0') == '1'
SHOW_DEVICE_WARNING = os.getenv('SHOW_DEVICE_WARNING', '1') == '1'
SHOW_ADVANCED_OPTIONS = os.getenv('SHOW_ADVANCED_OPTIONS', '1') == '1'
SHOW_UPSCALE_TO_256_BUTTON = os.getenv('SHOW_UPSCALE_TO_256_BUTTON',
'0') == '1'
SHOW_NUM_IMAGES = os.getenv('SHOW_NUM_IMAGES_OPTION', '1') == '1'
SHOW_CUSTOM_TIMESTEPS_1 = os.getenv('SHOW_CUSTOM_TIMESTEPS_1', '1') == '1'
SHOW_CUSTOM_TIMESTEPS_2 = os.getenv('SHOW_CUSTOM_TIMESTEPS_2', '1') == '1'
SHOW_NUM_STEPS_1 = os.getenv('SHOW_NUM_STEPS_1', '0') == '1'
SHOW_NUM_STEPS_2 = os.getenv('SHOW_NUM_STEPS_2', '0') == '1'
SHOW_NUM_STEPS_3 = os.getenv('SHOW_NUM_STEPS_3', '1') == '1'
GALLERY_COLUMN_NUM = int(os.getenv('GALLERY_COLUMN_NUM', '4'))
# Parameters
MAX_QUEUE_SIZE = int(os.getenv('MAX_QUEUE_SIZE', '10'))
MAX_SEED = np.iinfo(np.int32).max
MAX_NUM_IMAGES = int(os.getenv('MAX_NUM_IMAGES', '4'))
DEFAULT_NUM_IMAGES = min(MAX_NUM_IMAGES,
int(os.getenv('DEFAULT_NUM_IMAGES', '4')))
MAX_NUM_STEPS = int(os.getenv('MAX_NUM_STEPS', '200'))
DEFAULT_CUSTOM_TIMESTEPS_1 = os.getenv('DEFAULT_CUSTOM_TIMESTEPS_1',
'smart100')
DEFAULT_CUSTOM_TIMESTEPS_2 = os.getenv('DEFAULT_CUSTOM_TIMESTEPS_2', 'smart50')
DEFAULT_NUM_STEPS_3 = int(os.getenv('DEFAULT_NUM_STEPS_3', '40'))
# Model options
DISABLE_AUTOMATIC_CPU_OFFLOAD = os.getenv(
'DISABLE_AUTOMATIC_CPU_OFFLOAD') == '1'
DISABLE_SD_X4_UPSCALER = os.getenv('DISABLE_SD_X4_UPSCALER') == '1'
# Other options
RUN_GARBAGE_COLLECTION = os.getenv('RUN_GARBAGE_COLLECTION', '1') == '1'
DEBUG = os.getenv('DEBUG') == '1'
# Default options for the public demo
if os.getenv('IS_PUBLIC_DEMO') == '1':
# UI
SHOW_DUPLICATE_BUTTON = True
SHOW_NUM_STEPS_3 = False
SHOW_CUSTOM_TIMESTEPS_1 = False
SHOW_CUSTOM_TIMESTEPS_2 = False
SHOW_NUM_IMAGES = False
# parameters
DEFAULT_CUSTOM_TIMESTEPS_1 = 'smart50'
UPLOAD_RESULT_IMAGE = True
# model
DISABLE_AUTOMATIC_CPU_OFFLOAD = True
RUN_GARBAGE_COLLECTION = False
|