configuration_stacked.py

from transformers import PretrainedConfig
import torch

class ImpressoConfig(PretrainedConfig):
    model_type = "floret"

# Register the configuration with the transformers library
ImpressoConfig.register_for_auto_class()