Spaces:

vinthony
/

SadTalker

Running on A10G

vinthony commited on Apr 9, 2023

Commit

4142979

•

1 Parent(s): 2ecc379

Update src/face3d/extract_kp_videos.py

Files changed (1) hide show

src/face3d/extract_kp_videos.py CHANGED Viewed

@@ -40,6 +40,7 @@ class KeypointExtractor():
                     break
                 except RuntimeError as e:
                     if str(e).startswith('CUDA'):
                         print("Warning: out of memory, sleep for 1s")
                         time.sleep(1)
                     else:

                     break
                 except RuntimeError as e:
                     if str(e).startswith('CUDA'):
+                        print(e)
                         print("Warning: out of memory, sleep for 1s")
                         time.sleep(1)
                     else: