Spaces:
Runtime error
Runtime error
| diff --git a/model.py b/model.py | |
| index 0134c39..3a7826c 100755 | |
| --- a/model.py | |
| +++ b/model.py | |
| class Generator(nn.Module): | |
| style_dim, | |
| n_mlp, | |
| channel_multiplier=2, | |
| + additional_multiplier=2, | |
| blur_kernel=[1, 3, 3, 1], | |
| lr_mlp=0.01, | |
| ): | |
| class Generator(nn.Module): | |
| 512: 32 * channel_multiplier, | |
| 1024: 16 * channel_multiplier, | |
| } | |
| + if additional_multiplier > 1: | |
| + for k in list(self.channels.keys()): | |
| + self.channels[k] *= additional_multiplier | |
| self.input = ConstantInput(self.channels[4]) | |
| self.conv1 = StyledConv( | |
| class Generator(nn.Module): | |
| getattr(self.noises, f"noise_{i}") for i in range(self.num_layers) | |
| ] | |
| - if truncation < 1: | |
| + if truncation_latent is not None: | |
| style_t = [] | |
| for style in styles: | |
| diff --git a/op/fused_act.py b/op/fused_act.py | |
| index 5d46e10..bc522ed 100755 | |
| --- a/op/fused_act.py | |
| +++ b/op/fused_act.py | |
| -import os | |
| - | |
| import torch | |
| from torch import nn | |
| from torch.nn import functional as F | |
| 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): | |
| @staticmethod | |
| def forward(ctx, grad_output, out, bias, negative_slope, scale): | |
| diff --git a/op/upfirdn2d.py b/op/upfirdn2d.py | |
| index 67e0375..6c5840e 100755 | |
| --- a/op/upfirdn2d.py | |
| +++ b/op/upfirdn2d.py | |
| from collections import abc | |
| -import os | |
| import torch | |
| from torch.nn import functional as F | |
| 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): | |
| @staticmethod | |
| def forward( | |