Parcourir la source

Remove some unused imports

oobabooga il y a 2 ans
Parent
commit
ff9f649c0c
2 fichiers modifiés avec 0 ajouts et 4 suppressions
  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