Logan Zoellner commited on
Commit
5fe351a
β€’
1 Parent(s): 4523071

maybe if I just include stylegan directly?

Browse files
Files changed (3) hide show
  1. README.md +0 -12
  2. patch +0 -85
  3. 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

  • SHA256: 2973e030ad45dc61759f6dcb1e2ffdff7b2aefb16704ba0551e7b80f6d289a09
  • Pointer size: 132 Bytes
  • Size of remote file: 8.23 MB