srun --gres=gpu:1 --pty python -c "import unsloth"
🦥 Unsloth: Will patch your computer to enable 2x faster free finetuning.
Traceback (most recent call last):
File "<string>", line 1, in <module>
File "/home/***/env/lib/python3.12/site-packages/unsloth/__init__.py", line 261, in <module>
from .models import *
File "/home/***/env/lib/python3.12/site-packages/unsloth/models/__init__.py", line 15, in <module>
from .llama import FastLlamaModel
File "/home/***/env/lib/python3.12/site-packages/unsloth/models/llama.py", line 20, in <module>
from ._utils import *
File "/home/***/env/lib/python3.12/site-packages/unsloth/models/_utils.py", line 466, in <module>
exec(config, globals())
File "<string>", line 1, in <module>
NameError: name 'PreTrainedConfig' is not defined. Did you mean: 'PretrainedConfig'?
pip install --upgrade unsloth unsloth_zooYes, latest stableColaborKaggleor local / cloud: localnvidia-smi: 1A new commit in transformers (163601c6197e2addccd30d9bf450f664766f2e71) today breaks unsloth import, results in