Pārlūkot izejas kodu

Sort the imports

oobabooga 2 gadi atpakaļ
vecāks
revīzija
0f6708c471
2 mainītis faili ar 2 papildinājumiem un 1 dzēšanām
  1. 1 0
      modules/RWKV.py
  2. 1 1
      modules/text_generation.py

+ 1 - 0
modules/RWKV.py

@@ -16,6 +16,7 @@ os.environ["RWKV_CUDA_ON"] = '0' #  '1' : use CUDA kernel for seq mode (much fas
 from rwkv.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 
+
 class RWKVModel:
     def __init__(self):
         pass

+ 1 - 1
modules/text_generation.py

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