File size: 410 Bytes
59506b2 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers.models.codegen import CodeGenConfig
from .configuration_measurement_pred import MeasurementPredictorConfig
class CodeGenMeasurementPredictorConfig(MeasurementPredictorConfig, CodeGenConfig):
model_type = "codegen_mp"
def __init__(self, **kwargs):
kwargs["sensor_token_id"] = 42848
super().__init__(**kwargs)
def get_emb_dim(self):
return self.n_embd
|