coding-alt/IF

1import os2 3import numpy as np4 5HF_TOKEN = os.getenv('HF_TOKEN')6UPLOAD_REPO_ID = os.getenv('UPLOAD_REPO_ID')7UPLOAD_RESULT_IMAGE = os.getenv('UPLOAD_RESULT_IMAGE') == '1'8 9# UI options10SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON', '0') == '1'11SHOW_DEVICE_WARNING = os.getenv('SHOW_DEVICE_WARNING', '1') == '1'12SHOW_ADVANCED_OPTIONS = os.getenv('SHOW_ADVANCED_OPTIONS', '1') == '1'13SHOW_UPSCALE_TO_256_BUTTON = os.getenv('SHOW_UPSCALE_TO_256_BUTTON',14                                       '0') == '1'15SHOW_NUM_IMAGES = os.getenv('SHOW_NUM_IMAGES_OPTION', '1') == '1'16SHOW_CUSTOM_TIMESTEPS_1 = os.getenv('SHOW_CUSTOM_TIMESTEPS_1', '1') == '1'17SHOW_CUSTOM_TIMESTEPS_2 = os.getenv('SHOW_CUSTOM_TIMESTEPS_2', '1') == '1'18SHOW_NUM_STEPS_1 = os.getenv('SHOW_NUM_STEPS_1', '0') == '1'19SHOW_NUM_STEPS_2 = os.getenv('SHOW_NUM_STEPS_2', '0') == '1'20SHOW_NUM_STEPS_3 = os.getenv('SHOW_NUM_STEPS_3', '1') == '1'21GALLERY_COLUMN_NUM = int(os.getenv('GALLERY_COLUMN_NUM', '4'))22 23# Parameters24MAX_QUEUE_SIZE = int(os.getenv('MAX_QUEUE_SIZE', '10'))25MAX_SEED = np.iinfo(np.int32).max26MAX_NUM_IMAGES = int(os.getenv('MAX_NUM_IMAGES', '4'))27DEFAULT_NUM_IMAGES = min(MAX_NUM_IMAGES,28                         int(os.getenv('DEFAULT_NUM_IMAGES', '4')))29MAX_NUM_STEPS = int(os.getenv('MAX_NUM_STEPS', '200'))30DEFAULT_CUSTOM_TIMESTEPS_1 = os.getenv('DEFAULT_CUSTOM_TIMESTEPS_1',31                                       'smart100')32DEFAULT_CUSTOM_TIMESTEPS_2 = os.getenv('DEFAULT_CUSTOM_TIMESTEPS_2', 'smart50')33DEFAULT_NUM_STEPS_3 = int(os.getenv('DEFAULT_NUM_STEPS_3', '40'))34 35# Model options36DISABLE_AUTOMATIC_CPU_OFFLOAD = os.getenv(37    'DISABLE_AUTOMATIC_CPU_OFFLOAD') == '1'38DISABLE_SD_X4_UPSCALER = os.getenv('DISABLE_SD_X4_UPSCALER') == '1'39 40# Other options41RUN_GARBAGE_COLLECTION = os.getenv('RUN_GARBAGE_COLLECTION', '1') == '1'42DEBUG = os.getenv('DEBUG') == '1'43 44# Default options for the public demo45if os.getenv('IS_PUBLIC_DEMO') == '1':46    # UI47    SHOW_DUPLICATE_BUTTON = True48    SHOW_NUM_STEPS_3 = False49    SHOW_CUSTOM_TIMESTEPS_1 = False50    SHOW_CUSTOM_TIMESTEPS_2 = False51    SHOW_NUM_IMAGES = False52    # parameters53    DEFAULT_CUSTOM_TIMESTEPS_1 = 'smart50'54    UPLOAD_RESULT_IMAGE = True55    # model56    DISABLE_AUTOMATIC_CPU_OFFLOAD = True57    RUN_GARBAGE_COLLECTION = False58