Pārlūkot izejas kodu

Remove some unused imports

oobabooga 2 gadi atpakaļ
vecāks
revīzija
ff9f649c0c
2 mainītis faili ar 0 papildinājumiem un 4 dzēšanām
  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