update inf importing in common.py

colossalai/utils/common.py

@@ -8,7 +8,7 @@ from pathlib import Path
 from typing import Callable, Dict, List, Optional, Union
 
 import torch
-from torch._six import inf
+from torch import inf
 from torch.nn.parameter import Parameter
 
 try: