fluidflow 0.2.2__tar.gz → 0.2.4__tar.gz

{fluidflow-0.2.2 → fluidflow-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fluidflow
-Version: 0.2.2
+Version: 0.2.4
 Summary: FluidFlow: models and training utilities for flow matching and diffusion
 Author-email: David Ramos <david.ramos.archilla@upm.es>
 Requires-Python: >=3.8

{fluidflow-0.2.2 → fluidflow-0.2.4}/fluidFlow/dit.py

@@ -322,9 +322,8 @@ class DiT(nn.Module):
         self.t_embedder = TimestepEmbedder(hidden_size, bias=use_bias)
         self.y_embedder = ConditionEmbedder(cond_dim, hidden_size, class_dropout_prob)
         num_patches = self.x_embedder.num_patches
-        # Will use fixed sin-cos embedding:
+        
         print(f"Creating DiT with {num_patches} patches.")
-        self.pos_embed = nn.Parameter(torch.zeros(1, num_patches, hidden_size), requires_grad=False)
         block_class = partial(WindowBlock, window_size=window_size) if attn_type == "window" else DiTBlock
         self.blocks = nn.ModuleList(
             [
@@ -355,10 +354,11 @@ class DiT(nn.Module):
 
         # Initialize (and freeze) pos_embed by sin-cos embedding:
         pos_embed = get_1d_sincos_pos_embed(
-            self.pos_embed.shape[-1], 
+            self.hidden_size, 
             self.x_embedder.num_patches
         )
-        self.pos_embed.data.copy_(torch.from_numpy(pos_embed).float().unsqueeze(0))
+        # Will use fixed sin-cos embedding:
+        self.register_buffer("pos_embed", torch.from_numpy(pos_embed).float().unsqueeze(0))
 
         # Initialize patch_embed like nn.Linear (instead of nn.Conv2d):
         w = self.x_embedder.proj.weight.data

{fluidflow-0.2.2 → fluidflow-0.2.4}/fluidflow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fluidflow
-Version: 0.2.2
+Version: 0.2.4
 Summary: FluidFlow: models and training utilities for flow matching and diffusion
 Author-email: David Ramos <david.ramos.archilla@upm.es>
 Requires-Python: >=3.8

{fluidflow-0.2.2 → fluidflow-0.2.4}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "fluidflow"
 authors = [{ name = "David Ramos", email = "david.ramos.archilla@upm.es" }]
-version = "0.2.2"
+version = "0.2.4"
 description = "FluidFlow: models and training utilities for flow matching and diffusion"
 readme = "README.md"
 requires-python = ">=3.8"