Missing module: torch.utils.checkpoint

https://github.com/hiyouga/LLaMA-Efficient-Tuning/issues/937
Similar as https://github.com/huggingface/transformers/issues/9617

modeling_baichuan.py

@@ -8,6 +8,7 @@ from threading import Thread
 from typing import List, Optional, Tuple, Union
 
 import torch
+import torch.utils.checkpoint
 from torch import nn
 from torch.nn import CrossEntropyLoss
 from torch.nn import functional as F