LIA-X-fast

Paused

YaohuiW commited on Aug 8

Commit

157e5ea

verified ·

1 Parent(s): 3f7f6ae

Update gradio_tabs/animation.py

Files changed (1) hide show

gradio_tabs/animation.py CHANGED Viewed

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

 import numpy as np
 import imageio
+extensions_dir = "./torch_extension/"
+os.environ["TORCH_EXTENSIONS_DIR"] = extensions_dir
 from networks.generator import Generator