소스 검색

Remove some unused imports

oobabooga 2 년 전
부모
커밋
ff9f649c0c
2개의 변경된 파일0개의 추가작업 그리고 4개의 파일을 삭제
  1. 0 3
      modules/RWKV.py
  2. 0 1
      modules/text_generation.py

+ 0 - 3
modules/RWKV.py

@@ -1,10 +1,7 @@
 import os
-import time
-import types
 from pathlib import Path
 
 import numpy as np
-import torch
 
 import modules.shared as shared
 

+ 0 - 1
modules/text_generation.py

@@ -5,7 +5,6 @@ import time
 import numpy as np
 import torch
 import transformers
-from rwkv.utils import PIPELINE, PIPELINE_ARGS
 from tqdm import tqdm
 
 import modules.shared as shared