[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
30094835d5
commit
e1aa01d8a6
1 changed files with 6 additions and 1 deletions
|
|
@ -36,7 +36,12 @@ def _torch(*, cuda = False, xpu = False):
|
|||
)
|
||||
|
||||
|
||||
def _modules(monkeypatch, *, torch = None, unsloth = False):
|
||||
def _modules(
|
||||
monkeypatch,
|
||||
*,
|
||||
torch = None,
|
||||
unsloth = False,
|
||||
):
|
||||
"""Stub sys.modules so the gate sees a chosen torch / unsloth state."""
|
||||
mods = dict(sys.modules)
|
||||
mods.pop("unsloth", None)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue