No module named 'deepspeed.checkpoint.utils'
I am doing my first try to fine tune falcon7,import torch, einops
from datasets import load_dataset
from peft import LoraConfig
from transformers import (
    AutoModelForCausalLM,
    AutoTokenizer,
    BitsAndBytesConfig,
    AutoTokenizer,
    TrainingArguments
)
from peft.tuners.lora import LoraLayer

from trl import SFTTrainerbut I gotModuleNotFoundError: No module named 'deepspeed.checkpoint.utils'Im on aws sagemaker and using ml.c5.2x and 4xlarge. with torch 2 gpu optimisedcan anybody please tell me what am i doing wrong?Thanks
2023-06-28T12:05:01Z
cc@smangrul
2023-06-28T12:23:43Z
Hello, please install the latest version of DeepSpeed
2023-06-28T12:37:53Z
smangrul:DeepSpeedThank you, it is installed along with all this libraries:!pip install -q -U bitsandbytes
!pip install -q -U git+https://github.com/huggingface/transformers.git 
!pip install -q -U git+https://github.com/huggingface/peft.git
!pip install -q -U git+https://github.com/huggingface/accelerate.git
!pip install -q -U datasets
!pip install -q -U trl
!pip install -q -U einops
!pip install torch
!pip install torchvision
!pip install deepspeedI also used p3 and g4 instances to make sure gpu is available.
2023-06-28T12:43:59Z
zhp27:!pip install deepspeedTry doing-U, and let us know what your deepspeed version is
2023-06-28T13:14:51Z
Thank you, that did the trick!It is version 0.9.5 now. (was 0.6 before)now  I face another issue, telling me cannot import name ‘MODEL_FILE_PREFIX’ from ‘deepspeed.checkpoint.constants’
2023-06-28T13:21:41Z
restart the kernel and it is resolved, thanks