Spaces:
Running
on
A10G
Running
on
A10G
hangzhang-nlp
commited on
Commit
•
6ffc2df
1
Parent(s):
4671ce0
Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ model_config = cfg.model_cfg
|
|
65 |
model_config.device_8bit = args.gpu_id
|
66 |
model_cls = registry.get_model_class(model_config.arch)
|
67 |
model = model_cls.from_config(model_config).to('cuda:{}'.format(args.gpu_id))
|
68 |
-
|
69 |
vis_processor_cfg = cfg.datasets_cfg.webvid.vis_processor.train
|
70 |
vis_processor = registry.get_processor_class(vis_processor_cfg.name).from_config(vis_processor_cfg)
|
71 |
chat = Chat(model, vis_processor, device='cuda:{}'.format(args.gpu_id))
|
|
|
65 |
model_config.device_8bit = args.gpu_id
|
66 |
model_cls = registry.get_model_class(model_config.arch)
|
67 |
model = model_cls.from_config(model_config).to('cuda:{}'.format(args.gpu_id))
|
68 |
+
model.eval()
|
69 |
vis_processor_cfg = cfg.datasets_cfg.webvid.vis_processor.train
|
70 |
vis_processor = registry.get_processor_class(vis_processor_cfg.name).from_config(vis_processor_cfg)
|
71 |
chat = Chat(model, vis_processor, device='cuda:{}'.format(args.gpu_id))
|