up

Author: metascroy

torchchat/export.py

@@ -439,7 +439,8 @@ def main(args):
             tokenizer,
             max_seq_length=builder_args.max_seq_length,
             support_tensor_subclass=output_dso_path is None
-            and output_aoti_package_path is None,
+            and output_aoti_package_path is None
+            and output_pte_path is None,
         )
         model_to_pte = model
         model_to_dso = model