(self)
| 37 | self.llm = LLM(model=name, max_model_len=2048, **kwargs) |
| 38 | |
| 39 | def is_direct_completion(self) -> bool: |
| 40 | return self.force_base_prompt or self.tokenizer.chat_template is None |
| 41 | |
| 42 | def codegen( |
| 43 | self, prompt: str, do_sample: bool = True, num_samples: int = 200 |