Update my_model/KBVQA.py
Browse files- my_model/KBVQA.py +1 -1
my_model/KBVQA.py
CHANGED
@@ -15,7 +15,7 @@ class KBVQA():
|
|
15 |
self.kbvqa_model_name = "m7mdal7aj/fine_tunned_llama_2_merged"
|
16 |
self.quantization='4bit'
|
17 |
self.bnb_config = self.create_bnb_config()
|
18 |
-
self.max_context_window =
|
19 |
self.add_eos_token = False
|
20 |
self.trust_remote = False
|
21 |
self.use_fast = True
|
|
|
15 |
self.kbvqa_model_name = "m7mdal7aj/fine_tunned_llama_2_merged"
|
16 |
self.quantization='4bit'
|
17 |
self.bnb_config = self.create_bnb_config()
|
18 |
+
self.max_context_window = 4000
|
19 |
self.add_eos_token = False
|
20 |
self.trust_remote = False
|
21 |
self.use_fast = True
|