feat: 'README'

modeling_flame.py

@@ -600,10 +600,9 @@ class MixedEmbedding(nn.Module):
         # [batch_size, patch_num, patch_size]
         patches = x.unfold(dimension=-1, size=patch_len, step=patch_len)
 
-        proj = nn.Linear(patch_len, self.d_model, bias=False)
-        proj.weight.data = resample(old=self.patch_embedding.proj.weight.data, new_patch_len=patch_len)
+        resampled_weight = resample(old=self.patch_embedding.proj.weight.data, new_patch_len=patch_len)
 
-        patch_embedding = proj(patches)
+        patch_embedding = F.linear(patches, resampled_weight)
         time_delay_embedding = self.time_delay_embedding(x, period=patch_len)
         embedding = patch_embedding + time_delay_embedding
         return embedding