3DGen-Arena / constants.py
ZhangYuhan's picture
update serve
f88246b
raw
history blame
972 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")
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", "http://139.196.105.158")
LOG_SERVER_SUBDOAMIN = os.getenv("LOG_SERVER_SUBDIR", "3DGen-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"
GIF_SERVER = os.getenv("GIF_SERVER", f"{LOG_SERVER}/get_gif")
RGBA_SERVER = os.getenv("RGBA_SERVER", f"{LOG_SERVER}/get_rgba")
NUM_SIDES = 2
EVALUATE_DIMS = 5
PROMPT_NUM = 510
TEXT_PROMPT_PATH = "offline/prompts.json"
MAX_ATTEMPTS = 5
REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN", "yourKey")