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