Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

url: save -mu downloads to new cache location #7826

Merged
merged 4 commits into from
Jun 8, 2024
Merged

url: save -mu downloads to new cache location #7826

Changes from 2 commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
59f7e0d
url: save -mu download to new cache location
ochafik Jun 7, 2024
92ec18f
Merge remote-tracking branch 'origin/master' into cache-url
ochafik Jun 8, 2024
bc5acfd
url: fs_get_cache_file_path util
ochafik Jun 8, 2024
76612a2
url: tweak sig of fs_get_cache_file
ochafik Jun 8, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 11 additions & 8 deletions common/common.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -191,6 +191,15 @@ int32_t cpu_get_num_math() {
// CLI argument parsing
//

static std::string get_cached_file(const std::string & path) {
ngxson marked this conversation as resolved.
Show resolved Hide resolved
std::string cache_directory = fs_get_cache_directory();
const bool success = fs_create_directory_with_parents(cache_directory);
if (!success) {
throw std::runtime_error("failed to create cache directory: " + cache_directory);
}
return cache_directory + string_split(path, '/').back();
}

void gpt_params_handle_model_default(gpt_params & params) {
if (!params.hf_repo.empty()) {
// short-hand to avoid specifying --hf-file -> default it to --model
Expand All @@ -200,19 +209,13 @@ void gpt_params_handle_model_default(gpt_params & params) {
}
params.hf_file = params.model;
} else if (params.model.empty()) {
std::string cache_directory = fs_get_cache_directory();
const bool success = fs_create_directory_with_parents(cache_directory);
if (!success) {
throw std::runtime_error("failed to create cache directory: " + cache_directory);
}
params.model = cache_directory + string_split(params.hf_file, '/').back();
params.model = get_cached_file(params.hf_file);
}
} else if (!params.model_url.empty()) {
if (params.model.empty()) {
auto f = string_split(params.model_url, '#').front();
f = string_split(f, '?').front();
f = string_split(f, '/').back();
params.model = "models/" + f;
params.model = get_cached_file(f);
}
} else if (params.model.empty()) {
params.model = DEFAULT_MODEL_PATH;
Expand Down
Loading