(
self,
base_lm,
knn_generator,
lmbda: float = 0.25,
knn_temp: float = 1.0,
)
| 38 | """ |
| 39 | |
| 40 | def __init__( |
| 41 | self, |
| 42 | base_lm, |
| 43 | knn_generator, |
| 44 | lmbda: float = 0.25, |
| 45 | knn_temp: float = 1.0, |
| 46 | ): |
| 47 | super().__init__(base_lm.config) |
| 48 | |
| 49 | self.base_lm = base_lm |
| 50 | self.knn_generator = knn_generator |
| 51 | self.lmbda = float(lmbda) |
| 52 | self.knn_temp = float(knn_temp) |
| 53 | |
| 54 | # ------------------------------------------------------------------ # |
| 55 | # 1. forward() |
nothing calls this directly
no outgoing calls
no test coverage detected