Igor Kasianenko b3992f326d fix is_xpu import 2 月之前
..
__init__.py c51d3af6c9 Use auto instead of bf16 to determine model dtype; Add comments to describe the various use_fp16 and pure_bf16 options 2 月之前
config_utils.py ef99b32e48 update refs 3 月之前
dataset_utils.py 0ca612fff3 fix more imports 3 月之前
flop_utils.py c0b6de8632 rename_folder 3 月之前
fsdp_utils.py b3992f326d fix is_xpu import 2 月之前
memory_utils.py c0b6de8632 rename_folder 3 月之前
plot_metrics.py c0b6de8632 rename_folder 3 月之前
train_utils.py c51d3af6c9 Use auto instead of bf16 to determine model dtype; Add comments to describe the various use_fp16 and pure_bf16 options 2 月之前