roundup_power2_divisions not needed with newer pytorch versions (#3540)
* roundup_power2_divisions not needed with newer pytorch versions * remove typo * update qwen3.5 moe 35b-a3b yaml for 5090 * more bug fixes * fix tests to match updated trainer * don't use fa2 for hooks test * reset plugins on the instance * retry download * fix references to renamed axolotl_cfg property on trainer * Fix ref to trainer cfg
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
"""Shared pytest fixtures"""
|
||||
|
||||
import collections
|
||||
import functools
|
||||
import importlib
|
||||
import logging
|
||||
@@ -473,6 +474,18 @@ def temp_dir() -> Generator[str, None, None]:
|
||||
shutil.rmtree(_temp_dir)
|
||||
|
||||
|
||||
@pytest.fixture(scope="function", autouse=True)
|
||||
def reset_plugin_manager():
|
||||
from axolotl.integrations.base import PluginManager
|
||||
|
||||
yield
|
||||
PluginManager._cfg = None
|
||||
# Don't reset _instance to None — module-level PLUGIN_MANAGER references
|
||||
# in train.py, model.py, etc. would become stale
|
||||
if PluginManager._instance is not None:
|
||||
PluginManager._instance.plugins = collections.OrderedDict()
|
||||
|
||||
|
||||
@pytest.fixture(scope="function", autouse=True)
|
||||
def torch_manual_seed():
|
||||
torch.manual_seed(42)
|
||||
|
||||
Reference in New Issue
Block a user