Skip to content

Commit

Permalink
propogate trusted
Browse files Browse the repository at this point in the history
  • Loading branch information
chrishalcrow committed Sep 10, 2024
1 parent 7198d2e commit e10fe69
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/spikeinterface/curation/model_based_curation.py
Original file line number Diff line number Diff line change
Expand Up @@ -283,9 +283,9 @@ def load_model(model_folder=None, repo_id=None, model_name=None, trusted=None):
elif model_folder is not None and repo_id is not None:
raise ValueError("Please only provide one of 'model_folder' or 'repo_id'.")
elif model_folder is not None:
model, model_info = _load_model_from_folder(model_folder=model_folder, model_name=model_name)
model, model_info = _load_model_from_folder(model_folder=model_folder, model_name=model_name, trusted=trusted)
else:
model, model_info = _load_model_from_huggingface(repo_id=repo_id, model_name=model_name)
model, model_info = _load_model_from_huggingface(repo_id=repo_id, model_name=model_name, trusted=trusted)

return model, model_info

Expand Down

0 comments on commit e10fe69

Please sign in to comment.