Skip to content

Commit 1d53e2f

Browse files
committed
filename variable check in Quantization.from_string
1 parent 5a2bc0d commit 1d53e2f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ai_diffusion/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ class Quantization(Enum):
107107
def from_string(s: str, filename: str | None = None):
108108
if s == "svdq":
109109
return Quantization.svdq
110-
elif "qwen" in filename and "svdq" in filename:
110+
elif filename and "qwen" in filename and "svdq" in filename:
111111
return Quantization.svdq
112112
else:
113113
return Quantization.none

0 commit comments

Comments
 (0)