Hugging Face's logo

imone
/
pangu_2_6B

Text Generation
Transformers
PyTorch
gpt_pangu
custom_code
Model card Files Community
4

import torch.utils.checkpoint

#4
by IANZHU - opened
base: refs/heads/main
←
from: refs/pr/4
Discussion Files changed
+1
-0
IANZHU

Otherwise, there wii be AttributeError: module 'torch.utils' has no attribute 'checkpoint'

import torch.utils.checkpoint1baa6c87
Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment