浏览代码

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