only export WallXConfig from wall_x package to avoid peft import in CI

This commit is contained in:
Pepijn
2025-12-17 16:29:18 +01:00
committed by Michel Aractingi
parent 3434a5d5df
commit eadd4c0856
2 changed files with 4 additions and 8 deletions

View File

@@ -15,7 +15,5 @@
# limitations under the License.
from .configuration_wall_x import WallXConfig
from .modeling_wall_x import WallXPolicy
from .processor_wall_x import make_wall_x_pre_post_processors
__all__ = ["WallXConfig", "WallXPolicy", "make_wall_x_pre_post_processors"]
__all__ = ["WallXConfig"]