tianbaoxiexxx commited on
Commit
b15814b
1 Parent(s): 9da50da

Fix security issue pointed by Ivan

Browse files
Files changed (1) hide show
  1. generation/generator.py +1 -1
generation/generator.py CHANGED
@@ -16,7 +16,7 @@ class Generator(object):
16
 
17
  def __init__(self, args, keys=None):
18
  self.args = args
19
- self.keys = keys
20
  self.current_key_id = 0
21
 
22
  # if the args provided, will initialize with the prompt builder for full usage
 
16
 
17
  def __init__(self, args, keys=None):
18
  self.args = args
19
+ self.__keys = keys
20
  self.current_key_id = 0
21
 
22
  # if the args provided, will initialize with the prompt builder for full usage