update for single gpu

Author: yaojin3616

src/llamafactory/train/tuner.py

@@ -142,3 +142,7 @@ def export_model(args: Optional[Dict[str, Any]] = None) -> None:
 
     except Exception as e:
         logger.warning("Cannot save tokenizer, please copy the files manually: {}.".format(e))
+
+
+if __name__ == "__main__":
+    run_exp()

src/llmtuner/wrapers/export_model.py

@@ -1,4 +1,4 @@
-from llmtuner import export_model
+from llamafactory.train.tuner import export_model
 
 
 def main():