File size: 410 Bytes
3dbfcc9
 
 
 
 
 
 
 
 
 
 
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