Spaces:
Running
Running
Commit History
feat: restore weights on CPU
5f954fc
fix: position embedding for generate method
ebac379
fix: typo
68cc185
fix: load from checkpoint
44b7c3e
feat(modeling): simplify abstract_init
fa72aa7
feat: handle model parallel
1bb3269
feat(train): distributed_shampoo with pjit
cc34d07
feat(model): clean way to load on cpu
12f323d
feat: use_artifact if run existing
a5ed112
Tokenizer, config, model can be loaded from wandb.
7e48337
Pedro Cuenca
commited on
Change import order again.
2b2be9b
Pedro Cuenca
commited on
Fix import order to make isort happy.
64d99b2
Pedro Cuenca
commited on
Accept changes suggested by linter.
9f522b8
Pedro Cuenca
commited on
Never consider local dirs as remote wandb references.
08dd098
Pedro Cuenca
commited on
Store resolved path after loading model.
55a631d
Pedro Cuenca
commited on
Override from_pretrained to support wandb artifacts.
1023afa
Pedro Cuenca
commited on