소스 검색

remove unused import

lchu 1 년 전
부모
커밋
1cc9df19e6
1개의 변경된 파일0개의 추가작업 그리고 1개의 파일을 삭제
  1. 0 1
      llama_finetuning.py

+ 0 - 1
llama_finetuning.py

@@ -66,7 +66,6 @@ import torch
 import torch.nn as nn
 import torch.cuda.nccl as nccl
 import torch.distributed as dist
-from torch.distributed.fsdp._common_utils import _is_fsdp_flattened
 from transformers.models.llama.modeling_llama import LlamaDecoderLayer