40 lines
1009 B
C++
40 lines
1009 B
C++
#pragma once
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
// Ref: https://huggingface.co/docs/hub/local-cache.md
|
|
|
|
namespace hf_cache {
|
|
|
|
struct hf_file {
|
|
std::string path;
|
|
std::string url;
|
|
std::string local_path;
|
|
std::string final_path;
|
|
std::string oid;
|
|
std::string repo_id;
|
|
};
|
|
|
|
using hf_files = std::vector<hf_file>;
|
|
|
|
// Get files from HF API
|
|
hf_files get_repo_files(
|
|
const std::string & repo_id,
|
|
const std::string & token
|
|
);
|
|
|
|
hf_files get_cached_files(const std::string & repo_id = {});
|
|
|
|
// Download file.url -> file.local_path (blobs/<oid>), skipping if already present.
|
|
// Returns false on network failure. HTTPS requires CPPHTTPLIB_OPENSSL_SUPPORT.
|
|
bool download_file(const hf_file & file, const std::string & token);
|
|
|
|
// Create snapshot path (link or move/copy) and return it
|
|
std::string finalize_file(const hf_file & file);
|
|
|
|
// Remove the entire cached directory for a repo, returns true if removed
|
|
bool remove_cached_repo(const std::string & repo_id);
|
|
|
|
} // namespace hf_cache
|