| 4 | |
| 5 | @dataclass |
| 6 | class XftConfig: |
| 7 | max_seq_len: int = 4096 |
| 8 | beam_width: int = 1 |
| 9 | eos_token_id: int = -1 |
| 10 | pad_token_id: int = -1 |
| 11 | num_return_sequences: int = 1 |
| 12 | is_encoder_decoder: bool = False |
| 13 | padding: bool = True |
| 14 | early_stopping: bool = False |
| 15 | data_type: str = "bf16_fp16" |
| 16 | |
| 17 | |
| 18 | class XftModel: |
no outgoing calls
no test coverage detected
searching dependent graphs…