Skip to content

GitLab

Unverified Commit cbe34c55 authored by Rocky Duan's avatar Rocky Duan Committed by GitHub
Browse files

Fix ColoTensorSpec for py11 (#5440)

parent a7790a92
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
colossalai/legacy/tensor/tensor_spec.py
View file @ cbe34c55
from dataclasses import dataclass
from dataclasses import dataclass, field
from typing import Optional
from colossalai.legacy.tensor.distspec import DistPlacementPattern, _DistSpec
......@@ -17,5 +17,5 @@ class ColoTensorSpec:
"""
pg: ProcessGroup
dist_attr: Optional[_DistSpec] = _DistSpec(DistPlacementPattern.REPLICATE)
dist_attr: Optional[_DistSpec] = field(default_factory=lambda: _DistSpec(DistPlacementPattern.REPLICATE))
compute_attr: Optional[ComputeSpec] = None
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment