Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -329,11 +329,6 @@ class FluxLatentDINOFlow(nn.Module):
|
|
| 329 |
else:
|
| 330 |
dino_tokens = self._cached_dino_map
|
| 331 |
|
| 332 |
-
dino_map_resized = F.interpolate(dino_map, size=(self.grid_size, self.grid_size), mode='bilinear', align_corners=False)
|
| 333 |
-
dino_tokens = self.dino_adapter(dino_map_resized)
|
| 334 |
-
dino_tokens = dino_tokens.flatten(2).transpose(1, 2)
|
| 335 |
-
dino_tokens = dino_tokens + self.type_emb_dino
|
| 336 |
-
|
| 337 |
tokens = torch.cat([x_tokens, pixel_tokens, dino_tokens], dim=1)
|
| 338 |
t_emb = self.t_embedder(t)
|
| 339 |
|
|
|
|
| 329 |
else:
|
| 330 |
dino_tokens = self._cached_dino_map
|
| 331 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 332 |
tokens = torch.cat([x_tokens, pixel_tokens, dino_tokens], dim=1)
|
| 333 |
t_emb = self.t_embedder(t)
|
| 334 |
|