Spaces:
Sleeping
Sleeping
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
|