LIA-X-fast

Paused

YaohuiW commited on Aug 8

Commit

148fbca

verified ·

1 Parent(s): 157e5ea

Update gradio_tabs/vid_edit.py

Files changed (1) hide show

gradio_tabs/vid_edit.py CHANGED Viewed

@@ -7,8 +7,8 @@ import numpy as np
 import imageio
 from einops import rearrange
-# extensions_dir = "./torch_extension/"
-# os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 from networks.generator import Generator

 import imageio
 from einops import rearrange
+extensions_dir = "./torch_extension/"
+os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 from networks.generator import Generator