feat(gpt): do re-gen only on None manual_seed

2026-08-30 17:05:27 +08:00 · 2024-08-03 17:11:08 +08:00
parent 61add70490
commit c140a0b691
1 changed files with 1 additions and 1 deletions
@@ -617,7 +617,7 @@ class GPT(nn.Module):
                    "unexpected end at index %s",
                    str([unexpected_idx.item() for unexpected_idx in finish.nonzero()]),
                )
-                if ensure_non_empty:
+                if ensure_non_empty and manual_seed is None:
                    if show_tqdm:
                        pbar.close()
                    self.logger.warning("regenerate in order to ensure non-empty")