Explorar o código

Sort the imports

oobabooga %!s(int64=2) %!d(string=hai) anos
pai
achega
0f6708c471
Modificáronse 2 ficheiros con 2 adicións e 1 borrados
  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