Skip to content

Commit 554a1d2

Browse files
committed
Fix quantizer
1 parent 8c91ed1 commit 554a1d2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/main.cu

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -426,6 +426,8 @@ int main(int argc, char* argv[]) {
426426

427427
auto output_dir = program.get("--output");
428428
model.set_file_output(output_dir);
429+
for (auto quantizer : model.m_quantizers)
430+
quantizer.set_path(model.m_path.string());
429431

430432
if (auto previous = program.present("--previous")) {
431433
model.load_weights(*previous);

0 commit comments

Comments
 (0)