|
@@ -20,7 +20,7 @@ logging.basicConfig(
|
|
|
)
|
|
)
|
|
|
logger = logging.getLogger(__name__)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
-STATUS_FILE = "/tmp/training_status.json"
|
|
|
|
|
|
|
+STATUS_FILE = "/home/user/training_status.json"
|
|
|
|
|
|
|
|
|
|
|
|
|
def write_status(status: str, detail: str = "", progress: float = 0.0):
|
|
def write_status(status: str, detail: str = "", progress: float = 0.0):
|
|
@@ -29,7 +29,7 @@ def write_status(status: str, detail: str = "", progress: float = 0.0):
|
|
|
|
|
|
|
|
|
|
|
|
|
def merge_and_push(
|
|
def merge_and_push(
|
|
|
- adapter_path: str = "/tmp/qwen3-uncensored-lora/final_adapter",
|
|
|
|
|
|
|
+ adapter_path: str = "/home/user/output/final_adapter",
|
|
|
hub_model_id: str = "",
|
|
hub_model_id: str = "",
|
|
|
push_to_hub: bool = True,
|
|
push_to_hub: bool = True,
|
|
|
):
|
|
):
|
|
@@ -82,7 +82,7 @@ def merge_and_push(
|
|
|
# -----------------------------------------------------------------------
|
|
# -----------------------------------------------------------------------
|
|
|
# 3. Save merged model
|
|
# 3. Save merged model
|
|
|
# -----------------------------------------------------------------------
|
|
# -----------------------------------------------------------------------
|
|
|
- output_path = "/tmp/merged_model"
|
|
|
|
|
|
|
+ output_path = "/home/user/merged"
|
|
|
write_status("merging", "Saving merged model...", 0.6)
|
|
write_status("merging", "Saving merged model...", 0.6)
|
|
|
logger.info(f"Saving merged model to: {output_path}")
|
|
logger.info(f"Saving merged model to: {output_path}")
|
|
|
|
|
|
|
@@ -163,9 +163,7 @@ if __name__ == "__main__":
|
|
|
import argparse
|
|
import argparse
|
|
|
|
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser = argparse.ArgumentParser()
|
|
|
- parser.add_argument(
|
|
|
|
|
- "--adapter-path", default="/tmp/qwen3-uncensored-lora/final_adapter"
|
|
|
|
|
- )
|
|
|
|
|
|
|
+ parser.add_argument("--adapter-path", default="/home/user/output/final_adapter")
|
|
|
parser.add_argument(
|
|
parser.add_argument(
|
|
|
"--hub-model-id",
|
|
"--hub-model-id",
|
|
|
required=True,
|
|
required=True,
|