cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: torch_dtype -> dtype (fix deprecation warning)

Browse Source
This commit is contained in:
Christian Medina
2026-07-01 07:38:36 -04:00
parent f651ca030c
commit f0ee6bc9a2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
training/scripts/train.py
View File

@@ -36,7 +36,7 @@ def train(config_path):
print(f"Loading model: {config['base_model']}") print(f"Loading model: {config['base_model']}")
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
config["base_model"], config["base_model"],
torch_dtype=torch.bfloat16, # Convert FP8 -> bf16 dtype=torch.bfloat16, # Convert FP8 -> bf16
device_map="cpu", # Load to CPU first device_map="cpu", # Load to CPU first
trust_remote_code=True, trust_remote_code=True,
) )