Spaces:
Running
Running
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]
|