aapot
commited on
Commit
•
3f9f8e9
1
Parent(s):
4ebb03d
Fix
Browse files
EasyLM/models/llama/llama_train.py
CHANGED
|
@@ -29,6 +29,7 @@ FLAGS, FLAGS_DEF = mlxu.define_flags_with_default(
|
|
| 29 |
seed=42,
|
| 30 |
mesh_dim='1,-1,1',
|
| 31 |
dtype='fp32',
|
|
|
|
| 32 |
total_steps=10000,
|
| 33 |
load_llama_config='',
|
| 34 |
update_llama_config='',
|
|
|
|
| 29 |
seed=42,
|
| 30 |
mesh_dim='1,-1,1',
|
| 31 |
dtype='fp32',
|
| 32 |
+
param_dtype='fp32',
|
| 33 |
total_steps=10000,
|
| 34 |
load_llama_config='',
|
| 35 |
update_llama_config='',
|