Spaces:
Sleeping
Sleeping
s.lepage
commited on
Commit
·
014beea
1
Parent(s):
98e6bb9
Remove misleading comment.
Browse files- src/model.py +1 -1
src/model.py
CHANGED
|
@@ -154,7 +154,7 @@ class ConditionalViT(nn.Module):
|
|
| 154 |
def forward(self, imgs: torch.Tensor, c: torch.Tensor = None):
|
| 155 |
"""
|
| 156 |
imgs : Batch of images
|
| 157 |
-
c : category indices.
|
| 158 |
"""
|
| 159 |
|
| 160 |
x = self.conv1(imgs) # shape = [*, width, grid, grid]
|
|
|
|
| 154 |
def forward(self, imgs: torch.Tensor, c: torch.Tensor = None):
|
| 155 |
"""
|
| 156 |
imgs : Batch of images
|
| 157 |
+
c : category indices.
|
| 158 |
"""
|
| 159 |
|
| 160 |
x = self.conv1(imgs) # shape = [*, width, grid, grid]
|