github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-21 13:11:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't use torch._six (#2775)

Browse Source 
* Don't use `torch._six`

This is a private API which is gone after https://github.com/pytorch/pytorch/pull/94709

* Update common.py
This commit is contained in:
Nikita Shulga 2023-02-16 17:22:45 -08:00 committed by GitHub
parent a88bc828d5
commit 01066152f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/utils/common.py
View File

@ -11,7 +11,7 @@ from typing import Callable, Dict, List, Optional, Union
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from torch._six import inf from torch import inf
from torch.nn.parameter import Parameter from torch.nn.parameter import Parameter
from colossalai.constants import IS_TENSOR_PARALLEL, NUM_PARTITIONS, TENSOR_PARALLEL_ATTRIBUTES from colossalai.constants import IS_TENSOR_PARALLEL, NUM_PARTITIONS, TENSOR_PARALLEL_ATTRIBUTES

2
colossalai/zero/sharded_optim/_utils.py
View File

@ -3,7 +3,7 @@ from typing import Optional
import torch import torch
import torch.distributed as dist import torch.distributed as dist
from torch._six import inf from torch.six import inf
from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
from colossalai.tensor import ColoParameter from colossalai.tensor import ColoParameter