oobabooga преди 2 години
родител
ревизия
0f6708c471
променени са 2 файла, в които са добавени 2 реда и са изтрити 1 реда
  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.model import RWKV
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 from rwkv.utils import PIPELINE, PIPELINE_ARGS
 
 
+
 class RWKVModel:
 class RWKVModel:
     def __init__(self):
     def __init__(self):
         pass
         pass

+ 1 - 1
modules/text_generation.py

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