mirror of
https://github.com/facebookresearch/blt.git
synced 2025-09-15 00:29:43 +00:00
Improve HF compatibility
Summary: Test Plan:
This commit is contained in:
parent
1b67cbe022
commit
05b71e331b
8 changed files with 254 additions and 24 deletions
|
@ -34,7 +34,7 @@ def test_entropy_model():
|
|||
"bpe_tokenizer_path": BLT_DATA / "tokenizer_final_32k.minus_inf_ws.model"
|
||||
},
|
||||
)
|
||||
entropy_model = load_entropy_model(
|
||||
entropy_model, _ = load_entropy_model(
|
||||
BLT_DATA / "checkpoint_0100000_consolidated",
|
||||
os.path.join(
|
||||
BLT_DATA,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue