From 2f16ce309ad4b5a1f40b4d0ab38c2e5fd7887318 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 1 Mar 2023 12:33:09 -0300 Subject: [PATCH] Rename a variable --- modules/RWKV.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/RWKV.py b/modules/RWKV.py index 5e701bf0..98b11847 100644 --- a/modules/RWKV.py +++ b/modules/RWKV.py @@ -29,7 +29,7 @@ class RWKVModel: pipeline = PIPELINE(model, tokenizer_path.as_posix()) result = self() - result.model = pipeline + result.pipeline = pipeline return result def generate(self, context, token_count=20, temperature=1, top_p=1, alpha_frequency=0.25, alpha_presence=0.25, token_ban=[0], token_stop=[], callback=None): @@ -42,4 +42,4 @@ class RWKVModel: token_stop = token_stop ) - return self.model.generate(context, token_count=token_count, args=args, callback=callback) + return self.pipeline.generate(context, token_count=token_count, args=args, callback=callback)