This commit is contained in:
jiawenliu64 2025-04-09 14:55:30 -07:00
parent c8a0b110c0
commit 5ecedc12e7
2 changed files with 2 additions and 18 deletions

View file

@ -10,7 +10,7 @@ from typing import Callable, Optional
import torch
from fairscale.nn.model_parallel.initialize import get_model_parallel_rank
from torch import nn, Tensor
from torch import Tensor, nn
from torch.nn import functional as F
from ...datatypes import QuantizationMode