src/constants.py

"""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 = 7680
PRE_BIAS = False
INIT_NORMALISE_DICT = None