test

huggingface · Apr 25, 2024 · 56b4ce2 · 56b4ce2
1 parent 595d238
commit 56b4ce2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/src/autotrain/app_utils.py b/src/autotrain/app_utils.py
@@ -127,8 +127,6 @@ def run_training(params, task_id, local=False, wait=False):
     else:
         raise NotImplementedError
 
-    if not local:
-        params.project_name = "/tmp/model"
     params.save(output_dir=params.project_name)
     cmd = launch_command(params=params)
     cmd = [str(c) for c in cmd]

diff --git a/src/autotrain/trainers/dreambooth/__main__.py b/src/autotrain/trainers/dreambooth/__main__.py
@@ -28,14 +28,13 @@ def train(config):
     if config.model in utils.XL_MODELS:
         config.xl = True
 
-    if config.project_name == "/tmp/model":
-        snapshot_download(
-            repo_id=config.image_path,
-            local_dir=config.project_name,
-            token=config.token,
-            repo_type="dataset",
-        )
-        config.image_path = "/tmp/model/concept1/"
+    snapshot_download(
+        repo_id=config.image_path,
+        local_dir="/tmp/model",
+        token=config.token,
+        repo_type="dataset",
+    )
+    config.image_path = "/tmp/model/concept1/"
     if config.image_path == f"{config.project_name}/autotrain-data":
         config.image_path = os.path.join(config.image_path, "concept1")