issues
search
google-research
/
t5x
Apache License 2.0
2.57k
stars
296
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Prevent divide by zero error in loss computation.
#1574
copybara-service[bot]
closed
1 week ago
1
Getting repetitions after pre-training
#1573
tonyv
opened
2 weeks ago
0
Set `aggregate=False` under t5x.
#1572
copybara-service[bot]
closed
2 weeks ago
0
Revert `get_local_data` to original implementation.
#1571
copybara-service[bot]
opened
3 weeks ago
0
Set `aggregate` to False under SaveArgs within t5x.
#1570
copybara-service[bot]
closed
3 weeks ago
0
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#1569
copybara-service[bot]
opened
3 weeks ago
0
Fix bug in T5X CheckpointManager argument propagation when SaveBestCheckpoint is set through gin.
#1568
copybara-service[bot]
opened
3 weeks ago
1
Qlora with flan t5 issue - ValueError: Trying to set a tensor of shape torch.Size([4096, 4096])
#1567
JhonDan1999
opened
4 weeks ago
0
Propagate metric_name_to_monitor to Orbax CheckpointManager.
#1566
copybara-service[bot]
opened
1 month ago
0
remove dependency on old protobuf
#1565
copybara-service[bot]
closed
1 month ago
0
Make OrbaxCheckpointManagerInterface gin.configurable.
#1564
copybara-service[bot]
closed
1 month ago
0
depend on orbax-checkpoint >= 0.5
#1563
copybara-service[bot]
closed
1 month ago
0
internal
#1562
copybara-service[bot]
closed
1 month ago
0
Set `aggregate` to False under SaveArgs within t5x.
#1561
copybara-service[bot]
closed
3 weeks ago
0
Enable ocdbt in t5x by default.
#1560
copybara-service[bot]
opened
1 month ago
0
Fix sync_global_devices issue.
#1559
copybara-service[bot]
closed
1 month ago
0
End support for legacy Pax formats with directory structure:
#1558
copybara-service[bot]
closed
1 month ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#1557
copybara-service[bot]
closed
1 month ago
0
Delegate to BasePyTreeCheckpointHandler rather than inheriting from it.
#1556
copybara-service[bot]
closed
1 month ago
0
Fix UnboundLocalError: local variable 'checkpoint_manager' referenced before assignment.
#1555
copybara-service[bot]
closed
1 month ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#1554
copybara-service[bot]
closed
1 month ago
0
madlad often hallucinates while translating
#1553
dumengnan
opened
1 month ago
0
Make CheckpointManagerConstructor gin-configurable.
#1552
copybara-service[bot]
closed
2 months ago
0
change checkpointing to be compatible with new serialization
#1551
copybara-service[bot]
closed
2 months ago
1
Error trying to import t5x
#1550
osamja
closed
1 month ago
5
Fixes typo in the documentation for `predict_batch_with_aux`.
#1549
copybara-service[bot]
closed
2 months ago
0
Increase t5x AsyncCheckpointer timeout to 600s.
#1548
copybara-service[bot]
closed
2 months ago
0
Use newer Orbax CheckpointManager API in T5X. This change is expected to be a no-op. Also removes a test case that is already covered in Orbax core.
#1547
copybara-service[bot]
closed
2 months ago
0
[train.py] UnboundLocalError: local variable 'checkpoint_manager' referenced before assignment
#1546
gpupuck
closed
1 month ago
2
Add ability to use checkpoints with non-standard directory name prefix.
#1545
copybara-service[bot]
closed
1 month ago
1
E tensorflow/tsl/platform/cloud/curl_http_request.cc
#1544
chenyue-max
opened
2 months ago
0
Added MAXO-GEN models, namely, SAC, AGIQA-1k, AGIQA-3k and AIGCIQA-2023.
#1543
copybara-service[bot]
opened
2 months ago
1
Added MAXO-GEN models, namely, SAC, AGIQA-1k, AGIQA-3k and AIGCIQA-2023.
#1542
copybara-service[bot]
opened
2 months ago
1
Fix typos in t5x/export_lib.py create_fake_input
#1541
copybara-service[bot]
closed
2 months ago
1
Rename dpsgd to orbax.contrib END_PUBLIC
#1540
copybara-service[bot]
closed
1 month ago
0
Rename DifferentiallyPrivateAggregateState to `orbax.contrib`
#1539
copybara-service[bot]
closed
1 month ago
0
Fix typo in export_lib.save docstring
#1538
copybara-service[bot]
closed
2 months ago
1
Enable AsyncCheckpointer by default in t5x.
#1537
copybara-service[bot]
closed
2 months ago
0
Flash attention
#1536
peregilk
opened
2 months ago
0
jax.tree.map->jax.tree_map
#1535
jntdst
opened
3 months ago
1
Bug fix for loading checkpoints when:
#1534
copybara-service[bot]
closed
3 months ago
1
module 'jax' has no attribute 'tree'
#1533
jntdst
opened
3 months ago
1
Switch the deprecated jax.tree_map to jax.tree.map
#1532
copybara-service[bot]
closed
3 months ago
1
DNS test cloud issue.
#1531
copybara-service[bot]
opened
3 months ago
0
Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps
#1530
copybara-service[bot]
closed
3 months ago
0
Allow clu.data.dataset_iterator.DatasetIterator in addition to tf.data.Iterator
#1529
copybara-service[bot]
closed
3 months ago
0
Fix edge case `step=0` and simplify logic.
#1528
copybara-service[bot]
closed
3 months ago
0
Fix issue with dtype being dropped from RestoreArgs.
#1527
copybara-service[bot]
closed
3 months ago
0
Eliminate remaining seqio vocabulary usage in airio.
#1526
copybara-service[bot]
closed
3 months ago
0
Simplify tokenizer class structure.
#1525
copybara-service[bot]
closed
3 months ago
0
Next