shank commited on
Commit ·
77156dd
1
Parent(s): bdec91d
COMPUTE_DRIVE fix
Browse files- training/train_grpo.py +1 -2
training/train_grpo.py
CHANGED
|
@@ -84,8 +84,7 @@ if not args.test_local:
|
|
| 84 |
f"transformers={_pkg_ver('transformers')} "
|
| 85 |
f"trl={_pkg_ver('trl')} "
|
| 86 |
f"accelerate={_pkg_ver('accelerate')} "
|
| 87 |
-
f"peft={_pkg_ver('peft')}
|
| 88 |
-
f"dtype={COMPUTE_DTYPE}"
|
| 89 |
)
|
| 90 |
|
| 91 |
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
| 84 |
f"transformers={_pkg_ver('transformers')} "
|
| 85 |
f"trl={_pkg_ver('trl')} "
|
| 86 |
f"accelerate={_pkg_ver('accelerate')} "
|
| 87 |
+
f"peft={_pkg_ver('peft')}"
|
|
|
|
| 88 |
)
|
| 89 |
|
| 90 |
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|