[Fix bug]Add generation module in init.py file (#2608)

fangfangssj · web-flow · commit 01597fc94019 · 2025-09-26T16:23:51.000+08:00
diff --git a/paddleformers/__init__.py b/paddleformers/__init__.py
@@ -54,6 +54,7 @@
 modules = [
     "data",
     "datasets",
+    "generation",
     "nn",
     "mergekit",
     "ops",
@@ -72,6 +73,7 @@
     from . import (
         data,
         datasets,
+        generation,
         mergekit,
         nn,
         ops,
