mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-12-14 16:12:41 +00:00
[example] opt does not depend on Titans (#1811)
This commit is contained in:
@@ -32,9 +32,9 @@ import datasets
|
||||
import torch
|
||||
import torch.distributed as dist
|
||||
from accelerate.utils import set_seed
|
||||
from context import barrier_context
|
||||
from datasets import load_dataset
|
||||
from packaging import version
|
||||
from titans.utils import barrier_context
|
||||
from torch.utils.data import DataLoader
|
||||
from tqdm.auto import tqdm
|
||||
from utils import colo_memory_cap
|
||||
|
||||
Reference in New Issue
Block a user