Update to new version of base image

This commit is contained in:
2025-10-24 15:45:06 +08:00
parent ee04aead1e
commit fad74b701b
476 changed files with 1270 additions and 46 deletions

View File

@@ -132,7 +132,7 @@ class UnquantizedLinearMethod(LinearMethodBase):
layer: torch.nn.Module,
x: torch.Tensor,
bias: Optional[torch.Tensor] = None) -> torch.Tensor:
if (x.shape[0] == 16384 or x.shape[0] == 15360):
if (x.shape[0] == 8192 or x.shape[0] == 16384 or x.shape[0] == 15360):
if bias is None:
return x @ layer.weight.T
else:

View File

@@ -1,4 +1,4 @@
import re
import regex as re
from enum import Enum
from typing import Any, Dict, Iterable, Optional, Union