hangzhang-nlp commited on
Commit
6ffc2df
1 Parent(s): 4671ce0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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))