Skip to content

Commit b608f1e

Browse files
committed
Fix default HF safetensors filename/revision
1 parent 33e6d8d commit b608f1e

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,8 @@ seq_ints = sequence_to_af2_indices("ACDEFGHIK")
6363
`HuggingFaceApi.hf_hub_download`. By default it pulls:
6464

6565
- `repo_id = "facebook/esmfold_v1"`
66-
- `filename = "esm.safetensors"`
67-
- `revision = "main"`
66+
- `filename = "model.safetensors"`
67+
- `revision = "refs/pr/6"`
6868

6969
Downloaded files are cached by HuggingFaceApi in your Julia depot (via OhMyArtifacts).
7070
You can override the source if you want to point at a PR or a specific commit:

src/weights.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,8 @@ end
176176

177177
function load_ESM(;
178178
repo_id::AbstractString = "facebook/esmfold_v1",
179-
filename::AbstractString = "esm.safetensors",
180-
revision::AbstractString = "main",
179+
filename::AbstractString = "model.safetensors",
180+
revision::AbstractString = "refs/pr/6",
181181
cache::Bool = true,
182182
local_files_only::Bool = false,
183183
use_esm_attn_map::Bool = false,

0 commit comments

Comments
 (0)