m7mdal7aj commited on
Commit
7fd408f
1 Parent(s): 450d7b7

Update my_model/KBVQA.py

Browse files
Files changed (1) hide show
  1. 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 = 4096
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