Fix typos in some files (#3250) [skip ci]
This commit is contained in:
@@ -457,7 +457,7 @@ class PatchManager:
|
|||||||
and self.cfg.flash_attention
|
and self.cfg.flash_attention
|
||||||
and not self.inference
|
and not self.inference
|
||||||
):
|
):
|
||||||
# TODO(MengqingCao): split these patches seperately
|
# TODO(MengqingCao): split these patches separately
|
||||||
from axolotl.monkeypatch.llama_attn_hijack_flash import (
|
from axolotl.monkeypatch.llama_attn_hijack_flash import (
|
||||||
is_xformers_swiglu_available,
|
is_xformers_swiglu_available,
|
||||||
replace_llama_mlp_with_swiglu,
|
replace_llama_mlp_with_swiglu,
|
||||||
|
|||||||
@@ -144,7 +144,7 @@ def recursive_kill(process: subprocess.Popen):
|
|||||||
@pytest.mark.skip(reason="flaky vllm tests in modal")
|
@pytest.mark.skip(reason="flaky vllm tests in modal")
|
||||||
class TestGRPO:
|
class TestGRPO:
|
||||||
"""
|
"""
|
||||||
Test case for GRPO training using multilpe GPUs
|
Test case for GRPO training using multiple GPUs
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def _utils_write_yaml_and_rewards(self, cfg, temp_dir, suffix=""):
|
def _utils_write_yaml_and_rewards(self, cfg, temp_dir, suffix=""):
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ class TestPreprocess:
|
|||||||
"""test cases for preprocess"""
|
"""test cases for preprocess"""
|
||||||
|
|
||||||
def test_w_deepspeed(self, temp_dir):
|
def test_w_deepspeed(self, temp_dir):
|
||||||
"""make sure preproces doesn't choke when using deepspeed in the config"""
|
"""make sure preprocess doesn't choke when using deepspeed in the config"""
|
||||||
|
|
||||||
cfg = DictDefault(
|
cfg = DictDefault(
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user