Spaces:
Runtime error
Runtime error
File size: 519 Bytes
0d998a6 bd23dfe ffb8c37 bd23dfe 3b6ef01 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
"""Manage constants for the app.
"""
import os
import pathlib
import torch
ASSETS_FOLDER = pathlib.Path(__file__).parent.parent / "assets"
FIGURES_FOLER = pathlib.Path(__file__).parent.parent / "figures"
HF_TOKEN = os.getenv("HF_TOKEN")
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
MODEL_NAME = "lc0-10-4238.onnx"
SAE_CONFIG = "debug"
LAYER = 9
ACTIVATION_DIM = 256
DICTIONARY_SIZE = 2048
PRE_BIAS = False
INIT_NORMALISE_DICT = None
FEATURE_DATASET = "Xmaster6y/lczero-planning-features"
|