mirror of
https://github.com/deepseek-ai/DeepSeek-Coder.git
synced 2025-04-19 01:59:17 -04:00
Merge 78d0fd332a
into b7ba565956
This commit is contained in:
commit
1b2b3a8a8a
@ -143,7 +143,7 @@ def train():
|
||||
|
||||
model = transformers.AutoModelForCausalLM.from_pretrained(
|
||||
model_args.model_name_or_path,
|
||||
torch_dtype=torch.bfloat16
|
||||
torch_dtype=torch.float16
|
||||
)
|
||||
|
||||
if training_args.local_rank == 0:
|
||||
|
Loading…
Reference in New Issue
Block a user