From d430234d819b72153ae33a75263134dddf6ab2dc Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Wed, 16 Nov 2022 11:59:13 +0800
Subject: [PATCH] Update RIFE.py

---
 model/RIFE.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model/RIFE.py b/model/RIFE.py
index 6abf090..9af628a 100644
--- a/model/RIFE.py
+++ b/model/RIFE.py
@@ -13,7 +13,7 @@ from model.loss import *
 from model.laplacian import *
 from model.refine import *
 
-device = torch.device("cuda")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
 class Model:
     def __init__(self, local_rank=-1, arbitrary=False):