tonyassi commited on
Commit
9feffc3
·
1 Parent(s): e33a302

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -17,6 +17,7 @@ from cheetah.conversation.conversation_llama2 import Chat, CONV_VISION
17
  from cheetah.models import *
18
  from cheetah.processors import *
19
 
 
20
  def parse_args():
21
  parser = argparse.ArgumentParser(description="Demo")
22
  parser.add_argument("--cfg-path", required=True, help="path to configuration file.")
@@ -38,6 +39,7 @@ def setup_seeds(seed = 50):
38
  torch.cuda.manual_seed_all(seed)
39
  cudnn.benchmark = False
40
  cudnn.deterministic = True
 
41
 
42
  print('Initializing Chat')
43
  #args = parse_args()
 
17
  from cheetah.models import *
18
  from cheetah.processors import *
19
 
20
+ '''
21
  def parse_args():
22
  parser = argparse.ArgumentParser(description="Demo")
23
  parser.add_argument("--cfg-path", required=True, help="path to configuration file.")
 
39
  torch.cuda.manual_seed_all(seed)
40
  cudnn.benchmark = False
41
  cudnn.deterministic = True
42
+ '''
43
 
44
  print('Initializing Chat')
45
  #args = parse_args()