Vision-CAIR
commited on
Commit
·
ca9fe0c
1
Parent(s):
ebfc123
Update app.py
Browse files
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 |
# ========================================
|