yagizdevre commited on
Commit
5dc2a44
·
1 Parent(s): e07e60b
__pycache__/modeling_ministu.cpython-312.pyc CHANGED
Binary files a/__pycache__/modeling_ministu.cpython-312.pyc and b/__pycache__/modeling_ministu.cpython-312.pyc differ
 
modeling_ministu.py CHANGED
@@ -433,7 +433,7 @@ class MiniSTU(PreTrainedModel):
433
  seq_len=config.seq_len,
434
  K=config.num_eigh,
435
  use_hankel_L=config.use_hankel_L,
436
- device=device,
437
  )
438
 
439
  self.num_layers = config.num_layers
 
433
  seq_len=config.seq_len,
434
  K=config.num_eigh,
435
  use_hankel_L=config.use_hankel_L,
436
+ device=config.device,
437
  )
438
 
439
  self.num_layers = config.num_layers