3DGen-Arena / constants.py
ZhangYuhan's picture
update server
6296cf9
raw
history blame
982 Bytes
import os
from pathlib import Path
os.chdir(Path.cwd())
LOGDIR = os.getenv("LOGDIR", "./3DGen-Arena-logs/vote_log")
IMAGE_DIR = os.getenv("IMAGE_DIR", f"{LOGDIR}/images")
OFFLINE_DIR = "./offline"
OFFLINE_GIF_DIR = os.path.join(OFFLINE_DIR, "gifs")
SERVER_PORT = os.getenv("SERVER_PORT", 7860)
ROOT_PATH = os.getenv("ROOT_PATH", None)
ELO_RESULTS_DIR = os.getenv("ELO_RESULTS_DIR", "./arena_elo/results/latest")
LOG_SERVER = os.getenv("LOG_SERVER", "https://tigerai.ca")
LOG_SERVER_SUBDOAMIN = os.getenv("LOG_SERVER_SUBDIR", "GenAI-Arena-hf-logs")
LOG_SERVER_ADDR = os.getenv("LOG_SERVER_ADDR", f"{LOG_SERVER}/{LOG_SERVER_SUBDOAMIN}")
# LOG SERVER API ENDPOINTS
APPEND_JSON = "append_json"
SAVE_IMAGE = "save_image"
SAVE_LOG = "save_log"
NUM_SIDES = 2
EVALUATE_DIMS = 5
TEXT_PROMPT_PATH = "offline/prompts_110.json"
IMAGE_PROMPT_PATH = "offline/image_urls.txt"
MAX_ATTEMPTS = 5
REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN", "r8_0BaoQW0G8nWFXY8YWBCCUDurANxCtY72rarv9")