Skip to content

Commit 4b41826

Browse files
authored
fix log error (#843)
Signed-off-by: n1ck-guo <heng.guo@intel.com>
1 parent 77c608f commit 4b41826

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

auto_round/compressors/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -823,7 +823,7 @@ def remove_duplicates(lst):
823823

824824
if isinstance(self.scheme, str) and self.scheme.lower().startswith("gguf"):
825825
for i in range(len(formats)):
826-
if formats[i] != "fake" and formats[i] != self.scheme.lower().startswith("gguf"):
826+
if formats[i] != "fake" and formats[i] != self.scheme.lower():
827827
logger.warning(
828828
f"reset format {formats[i]} to {self.scheme.lower()} "
829829
f"since scheme {self.scheme} can only be exported to format {self.scheme.lower()}"

auto_round/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1385,7 +1385,6 @@ def check_and_mark_fp8_model(model: torch.nn.Module) -> bool:
13851385
if _is_fp8_linear(m):
13861386
m.is_fp8_linear = True
13871387
if not hasattr(model, "is_fp8"):
1388-
logger.warning("the support for fp8 model as input is experimental, please use with caution.")
13891388
model.is_fp8 = True
13901389
if hasattr(model, "is_fp8"):
13911390
return True

0 commit comments

Comments
 (0)