chong.zhang commited on
Commit
c1e0f6f
·
1 Parent(s): 2f1a479
Files changed (1) hide show
  1. inspiremusic/cli/inference.py +1 -1
inspiremusic/cli/inference.py CHANGED
@@ -54,7 +54,7 @@ class InspireMusicUnified:
54
 
55
  # Set model_dir or default to downloading if it doesn't exist
56
  if model_dir is None:
57
- model_dir = f"{download_model_dir}/{model_name}"
58
  print(model_dir)
59
  download_model_dir = os.path.dirname(model_dir)
60
  print(download_model_dir)
 
54
 
55
  # Set model_dir or default to downloading if it doesn't exist
56
  if model_dir is None:
57
+ model_dir = f"../../pretrained_models/{model_name}"
58
  print(model_dir)
59
  download_model_dir = os.path.dirname(model_dir)
60
  print(download_model_dir)