seawolf2357 commited on
Commit
feda343
·
verified ·
1 Parent(s): e8a44f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ model = load_model_checkpoint(model, ckpt_path)
68
  model.eval()
69
 
70
  # 모델을 DataParallel로 감싸서 여러 GPU에서 실행 가능하게 설정
71
- model = torch.nn.DataParallel(model)
72
  model = model.cuda()
73
 
74
 
 
68
  model.eval()
69
 
70
  # 모델을 DataParallel로 감싸서 여러 GPU에서 실행 가능하게 설정
71
+ #model = torch.nn.DataParallel(model)
72
  model = model.cuda()
73
 
74