fix bug in time embedding

model.py

@@ -28,8 +28,8 @@ class Sinusoidal_embedding(torch.nn.Module):
 #Time Embeddings 
 class Time_embeddings(torch.nn.Module):
     def __init__(self, in_dim:int, out_dim:int):
-        super().__init__
-        self.Sinusoidal_waves = Sinusoidal_embedding(dim=in_dim)
+        super().__init__()
+        self.Sinusoidal_waves = Sinusoidal_embedding(in_dim)
 
         self.mlp = nn.Sequential(
             nn.Linear(in_dim, out_dim),