Skip to content

Commit 2544104

Browse files
authored
Fix typo in generate (#1520)
1 parent efcbdab commit 2544104

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

recipes/generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,7 @@ def generate(self, cfg: DictConfig):
163163
logger.info(f"Warmup run for quantized model takes: {t:.02f} sec")
164164

165165
t0 = time.perf_counter()
166-
generated_tokens = generate(
166+
generated_tokens = generation.generate(
167167
model=self._model,
168168
prompt=prompt,
169169
max_generated_tokens=cfg.max_new_tokens,

0 commit comments

Comments
 (0)