Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|