(self, prompt)
| 30 | print(f'Memory footprint: {self.model.get_memory_footprint() / 1e6:.2f} MB') |
| 31 | |
| 32 | def format_prompt(self, prompt): |
| 33 | return f'Instructions: You are Helper, a large language model full of intelligence. Respond conversationally.</s>User: {prompt}</s> Helper: ' |
| 34 | |
| 35 | @retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6)) |
| 36 | def generate_text(self, prompt): |