Skip to content

Commit ece4d74

Browse files
committed
added qlinear
1 parent 9c02a92 commit ece4d74

File tree

2 files changed

+436
-0
lines changed

2 files changed

+436
-0
lines changed

auto_round_extension/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,3 +11,8 @@
1111
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
14+
15+
from auto_round_extension.qlinear import QuantLinear, QuantLinearGPTQ, QuantLinearAWQ
16+
qlinear_classes = (QuantLinear, QuantLinearGPTQ)
17+
18+
awq_classes = (QuantLinearAWQ,)

0 commit comments

Comments
 (0)