Spaces:

100XZX001
/

code-review-training

Sleeping

100XZX001 commited on Apr 25

Commit

d9cfe52

verified ·

1 Parent(s): 45d29c0

Update training.py

Files changed (1) hide show

training.py CHANGED Viewed

@@ -1,7 +1,7 @@
 # training.py – Memory‑safe: Phi‑3‑mini + Expert Demos + Fast PPO (2 iterations)
 import os
 os.environ["TRITON_DISABLE"] = "1"
-os.environ["TOKENIZERS_PARALLELISM"] = "false""  # stay in Python     # Issue #12: prevent OOM from parallel tokenization
 import torch._dynamo
 torch._dynamo.config.disable = True

 # training.py – Memory‑safe: Phi‑3‑mini + Expert Demos + Fast PPO (2 iterations)
 import os
 os.environ["TRITON_DISABLE"] = "1"
+os.environ["TOKENIZERS_PARALLELISM"] = "false"
 import torch._dynamo
 torch._dynamo.config.disable = True