Hazan-Lab
/

Flash_STU_550M

Model card Files Files and versions

yagizdevre commited on Jan 30, 2025

Commit

e07e60b

·

1 Parent(s): 5dd9ee6

typo fix

Files changed (1) hide show

modeling_ministu.py +1 -2

modeling_ministu.py CHANGED Viewed

@@ -433,8 +433,7 @@ class MiniSTU(PreTrainedModel):
             seq_len=config.seq_len,
             K=config.num_eigh,
             use_hankel_L=config.use_hankel_L,
-            device=device=device,
-            dtype=torch_dtype,
         )
         self.num_layers = config.num_layers

             seq_len=config.seq_len,
             K=config.num_eigh,
             use_hankel_L=config.use_hankel_L,
+            device=device,
         )
         self.num_layers = config.num_layers