Spaces:
Runtime error
Runtime error
Logan Zoellner
commited on
Commit
Β·
5fe351a
1
Parent(s):
4523071
maybe if I just include stylegan directly?
Browse files- README.md +0 -12
- patch +0 -85
- samples/sample.jpg +0 -3
README.md
DELETED
@@ -1,12 +0,0 @@
|
|
1 |
-
---
|
2 |
-
title: TADNE
|
3 |
-
emoji: π
|
4 |
-
colorFrom: green
|
5 |
-
colorTo: red
|
6 |
-
sdk: gradio
|
7 |
-
sdk_version: 3.0.2
|
8 |
-
app_file: app.py
|
9 |
-
pinned: false
|
10 |
-
---
|
11 |
-
|
12 |
-
Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
patch
DELETED
@@ -1,85 +0,0 @@
|
|
1 |
-
diff --git a/model.py b/model.py
|
2 |
-
index 0134c39..3a7826c 100755
|
3 |
-
--- a/model.py
|
4 |
-
+++ b/model.py
|
5 |
-
@@ -395,6 +395,7 @@ class Generator(nn.Module):
|
6 |
-
style_dim,
|
7 |
-
n_mlp,
|
8 |
-
channel_multiplier=2,
|
9 |
-
+ additional_multiplier=2,
|
10 |
-
blur_kernel=[1, 3, 3, 1],
|
11 |
-
lr_mlp=0.01,
|
12 |
-
):
|
13 |
-
@@ -426,6 +427,9 @@ class Generator(nn.Module):
|
14 |
-
512: 32 * channel_multiplier,
|
15 |
-
1024: 16 * channel_multiplier,
|
16 |
-
}
|
17 |
-
+ if additional_multiplier > 1:
|
18 |
-
+ for k in list(self.channels.keys()):
|
19 |
-
+ self.channels[k] *= additional_multiplier
|
20 |
-
|
21 |
-
self.input = ConstantInput(self.channels[4])
|
22 |
-
self.conv1 = StyledConv(
|
23 |
-
@@ -518,7 +522,7 @@ class Generator(nn.Module):
|
24 |
-
getattr(self.noises, f"noise_{i}") for i in range(self.num_layers)
|
25 |
-
]
|
26 |
-
|
27 |
-
- if truncation < 1:
|
28 |
-
+ if truncation_latent is not None:
|
29 |
-
style_t = []
|
30 |
-
|
31 |
-
for style in styles:
|
32 |
-
diff --git a/op/fused_act.py b/op/fused_act.py
|
33 |
-
index 5d46e10..bc522ed 100755
|
34 |
-
--- a/op/fused_act.py
|
35 |
-
+++ b/op/fused_act.py
|
36 |
-
@@ -1,5 +1,3 @@
|
37 |
-
-import os
|
38 |
-
-
|
39 |
-
import torch
|
40 |
-
from torch import nn
|
41 |
-
from torch.nn import functional as F
|
42 |
-
@@ -7,16 +5,6 @@ from torch.autograd import Function
|
43 |
-
from torch.utils.cpp_extension import load
|
44 |
-
|
45 |
-
|
46 |
-
-module_path = os.path.dirname(__file__)
|
47 |
-
-fused = load(
|
48 |
-
- "fused",
|
49 |
-
- sources=[
|
50 |
-
- os.path.join(module_path, "fused_bias_act.cpp"),
|
51 |
-
- os.path.join(module_path, "fused_bias_act_kernel.cu"),
|
52 |
-
- ],
|
53 |
-
-)
|
54 |
-
-
|
55 |
-
-
|
56 |
-
class FusedLeakyReLUFunctionBackward(Function):
|
57 |
-
@staticmethod
|
58 |
-
def forward(ctx, grad_output, out, bias, negative_slope, scale):
|
59 |
-
diff --git a/op/upfirdn2d.py b/op/upfirdn2d.py
|
60 |
-
index 67e0375..6c5840e 100755
|
61 |
-
--- a/op/upfirdn2d.py
|
62 |
-
+++ b/op/upfirdn2d.py
|
63 |
-
@@ -1,5 +1,4 @@
|
64 |
-
from collections import abc
|
65 |
-
-import os
|
66 |
-
|
67 |
-
import torch
|
68 |
-
from torch.nn import functional as F
|
69 |
-
@@ -7,16 +6,6 @@ from torch.autograd import Function
|
70 |
-
from torch.utils.cpp_extension import load
|
71 |
-
|
72 |
-
|
73 |
-
-module_path = os.path.dirname(__file__)
|
74 |
-
-upfirdn2d_op = load(
|
75 |
-
- "upfirdn2d",
|
76 |
-
- sources=[
|
77 |
-
- os.path.join(module_path, "upfirdn2d.cpp"),
|
78 |
-
- os.path.join(module_path, "upfirdn2d_kernel.cu"),
|
79 |
-
- ],
|
80 |
-
-)
|
81 |
-
-
|
82 |
-
-
|
83 |
-
class UpFirDn2dBackward(Function):
|
84 |
-
@staticmethod
|
85 |
-
def forward(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
samples/sample.jpg
DELETED
Git LFS Details
|