Vision-CAIR commited on
Commit
ca9fe0c
·
1 Parent(s): ebfc123

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -19,9 +19,6 @@ from minigpt4.processors import *
19
  from minigpt4.runners import *
20
  from minigpt4.tasks import *
21
 
22
- print(f"Is CUDA available: {torch.cuda.is_available()}")
23
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
24
-
25
  def parse_args():
26
  parser = argparse.ArgumentParser(description="Demo")
27
  parser.add_argument("--cfg-path", type=str, default='eval_configs/minigpt4.yaml', help="path to configuration file.")
@@ -62,6 +59,8 @@ vis_processor = registry.get_processor_class(vis_processor_cfg.name).from_config
62
  chat = Chat(model, vis_processor)
63
  print('Initialization Finished')
64
 
 
 
65
  # ========================================
66
  # Gradio Setting
67
  # ========================================
 
19
  from minigpt4.runners import *
20
  from minigpt4.tasks import *
21
 
 
 
 
22
  def parse_args():
23
  parser = argparse.ArgumentParser(description="Demo")
24
  parser.add_argument("--cfg-path", type=str, default='eval_configs/minigpt4.yaml', help="path to configuration file.")
 
59
  chat = Chat(model, vis_processor)
60
  print('Initialization Finished')
61
 
62
+ exit()
63
+
64
  # ========================================
65
  # Gradio Setting
66
  # ========================================