mirror of
https://github.com/deepseek-ai/DeepSeek-Coder
synced 2025-06-26 18:25:53 +00:00
Merge 78d0fd332a into b7ba565956
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user