edit gradio_demo.py

FONT/gradio_demo.py

@@ -249,9 +249,11 @@ class FONT():
 
             for j in range(len(kp_driving_all)):
                 kp_driving_all[j]['value'] = one_euro_filter_v.process(kp_driving_all[j]['value'].cpu() * 10) / 10
-                kp_driving_all[j]['value'] = kp_driving_all[j]['value'].cuda()
+                if self.device == 'cuda':
+                    kp_driving_all[j]['value'] = kp_driving_all[j]['value'].cuda()
                 kp_driving_all[j]['jacobian'] = one_euro_filter_j.process(kp_driving_all[j]['jacobian'].cpu() * 10) / 10
-                kp_driving_all[j]['jacobian'] = kp_driving_all[j]['jacobian'].cuda()
+                if self.device == 'cuda':
+                    kp_driving_all[j]['jacobian'] = kp_driving_all[j]['jacobian'].cuda()
 
             for frame_idx in tqdm(range(len(deco_out[0]))):