Skip to content

GitLab

"docs/source/en/using-diffusers/custom_pipeline_examples.mdx" did not exist on "52e8fdb8aec7c3676f2e6987291d16167aad9f4e"
Unverified Commit a8318353 authored by pollyyan's avatar pollyyan Committed by GitHub
Browse files

Merge pull request #5936 from LDOUBLEV/dygraph 

fix slim export model
parents d76bc57a 1c589d6a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
deploy/slim/quantization/export_model.py
View file @ a8318353
...@@ -133,6 +133,7 @@ def main(): ...@@ -133,6 +133,7 @@ def main():
sub_model_save_path, logger) sub_model_save_path, logger)
else: else:
save_path = os.path.join(save_path, "inference") save_path = os.path.join(save_path, "inference")
model.eval()
export_single_model(quanter, model, infer_shape, save_path, logger) export_single_model(quanter, model, infer_shape, save_path, logger)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment