config.py

import os
from cached_path import cached_path
# Configurations for inference
INFERENCE_WEIGHTS_PATH = str(cached_path('hf://ElectricAlexis/NotaGen/weights_notagenx_p_size_16_p_length_1024_p_layers_20_h_size_1280.pth'))               # Path to weights for inference# Folder to save output files
TOP_K = 9                                                       # Top k for sampling
TOP_P = 0.9                                                      # Top p for sampling
TEMPERATURE = 1.2                                                 # Temperature for sampling

# Configurations for model
PATCH_STREAM = True                                             # Stream training / inference
PATCH_SIZE = 16                                                # Patch Size
PATCH_LENGTH = 1024                                             # Patch Length
CHAR_NUM_LAYERS = 6                                             # Number of layers in the decoder
PATCH_NUM_LAYERS = 20                                           # Number of layers in the encoder
HIDDEN_SIZE = 1280                                               # Hidden Size