ColossalAI/colossalai/fx/tracer/bias_addition_patch/__init__.py

3 lines
90 B
Python

from .patched_bias_addition_function import *
from .patched_bias_addition_module import *