SFTTrainer - psutil is not defined

[cghall19]

I am trying to run some fairly simple examples, but when running the SFTTrainer it's throwing a psutil is not defined deep in the SFTTrainer. The exact same thing happens running locally and using google collab.

--> 854 train_dataset = self._prepare_dataset(
855 train_dataset, processing_class, args, args.packing, formatting_func, "train"
856 )
857 if eval_dataset is not None:
858 packing = args.packing if args.eval_packing is None else args.eval_packing

File ~/code/FineTuning/aviation/unsloth_compiled_cache/UnslothSFTTrainer.py:1026, in _UnslothSFTTrainer._prepare_dataset(self, dataset, processing_class, args, packing, formatting_func, dataset_name)
1024 dataset_num_proc = getattr(args, "dataset_num_proc", None)
1025 if dataset_num_proc is None:
-> 1026 dataset_num_proc = max(psutil.cpu_count()+4, 2)
1027 # Check memory left so we can reduce multiprocessing to converse memory
1028 memory_gb_left = psutil.virtual_memory().available / (1024**3)

NameError: name 'psutil' is not defined

Not sure where to go with this? I have tried all versions from 19.1 upwards

Thanks
Chris

[cghall19]

Found the issue, it didn't like the use of args = training arguments in my SFTTrainer definition, had to switch to use SFTConfig.