Update

.pre-commit-config.yaml

@@ -1,4 +1,4 @@
-exclude: ^DualStyleGAN
+exclude: ^(DualStyleGAN|patch)
 repos:
 - repo: https://github.com/pre-commit/pre-commit-hooks
   rev: v4.2.0

dualstylegan.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import argparse
 import os
 import pathlib
+import subprocess
 import sys
 from typing import Callable, Union
 
@@ -14,9 +15,9 @@ import torch
 import torch.nn as nn
 import torchvision.transforms as T
 
-if os.environ.get('SYSTEM') == 'spaces':
-    os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/fused_act.py")
-    os.system("sed -i '10,17d' DualStyleGAN/model/stylegan/op/upfirdn2d.py")
+if os.getenv('SYSTEM') == 'spaces':
+    with open('patch') as f:
+        subprocess.run('patch -p1'.split(), cwd='DualStyleGAN', stdin=f)
 
 app_dir = pathlib.Path(__file__).parent
 submodule_dir = app_dir / 'DualStyleGAN'

patch

@@ -0,0 +1,38 @@
+diff --git a/model/stylegan/op/fused_act.py b/model/stylegan/op/fused_act.py
+index c3b735c..67e7972 100644
+--- a/model/stylegan/op/fused_act.py
++++ b/model/stylegan/op/fused_act.py
+@@ -7,14 +7,6 @@ from torch.autograd import Function
+ from torch.utils.cpp_extension import load
+ 
+ 
+-module_path = os.path.dirname(__file__)
+-fused = load(
+-    "fused",
+-    sources=[
+-        os.path.join(module_path, "fused_bias_act.cpp"),
+-        os.path.join(module_path, "fused_bias_act_kernel.cu"),
+-    ],
+-)
+ 
+ 
+ class FusedLeakyReLUFunctionBackward(Function):
+diff --git a/model/stylegan/op/upfirdn2d.py b/model/stylegan/op/upfirdn2d.py
+index 67e0375..97565d7 100644
+--- a/model/stylegan/op/upfirdn2d.py
++++ b/model/stylegan/op/upfirdn2d.py
+@@ -7,14 +7,6 @@ from torch.autograd import Function
+ from torch.utils.cpp_extension import load
+ 
+ 
+-module_path = os.path.dirname(__file__)
+-upfirdn2d_op = load(
+-    "upfirdn2d",
+-    sources=[
+-        os.path.join(module_path, "upfirdn2d.cpp"),
+-        os.path.join(module_path, "upfirdn2d_kernel.cu"),
+-    ],
+-)
+ 
+ 
+ class UpFirDn2dBackward(Function):