| 21 | |
| 22 | @dataclass |
| 23 | class GenerationConfig: |
| 24 | max_length: Optional[int] = None |
| 25 | top_p: Optional[float] = None |
| 26 | temperature: Optional[float] = None |
| 27 | do_sample: Optional[bool] = True |
| 28 | repetition_penalty: Optional[float] = 1.0 |
| 29 | |
| 30 | |
| 31 | def load_vector_db(): |
no outgoing calls
no test coverage detected