LiamKhoaLe commited on
Commit
1db513a
·
1 Parent(s): e7283f8

Migrate snapshot folder files to parent

Browse files
Files changed (1) hide show
  1. download_model.py +1 -1
download_model.py CHANGED
@@ -16,7 +16,7 @@ print("Model path: ", snapshots_dir)
16
  if os.path.exists(snapshots_dir):
17
  snapshot_folders = os.listdir(snapshots_dir)
18
  if snapshot_folders:
19
- # snapshot_dir = os.path.join(snapshots_dir, snapshot_folders[0]) # Get first snapshot folder (e.g., migrate config file)
20
  # Move all files and directories from the snapshot folder to /app/model_cache
21
  for item in os.listdir(snapshot_folders):
22
  source_path = os.path.join(snapshot_folders, item)
 
16
  if os.path.exists(snapshots_dir):
17
  snapshot_folders = os.listdir(snapshots_dir)
18
  if snapshot_folders:
19
+ snapshot_dir = os.path.join(snapshots_dir, snapshot_folders[0]) # Get first snapshot folder (e.g., migrate config file)
20
  # Move all files and directories from the snapshot folder to /app/model_cache
21
  for item in os.listdir(snapshot_folders):
22
  source_path = os.path.join(snapshot_folders, item)