ShaswatRobotics
/

world_model_test

Model card Files Files and versions

ShaswatRobotics commited on Dec 29, 2025

Commit

0030af2

·

verified ·

1 Parent(s): ee3f4d0

Update iris/src/tokenizer.py

Files changed (1) hide show

iris/src/tokenizer.py +2 -2

iris/src/tokenizer.py CHANGED Viewed

@@ -8,8 +8,8 @@ from einops import rearrange
 import torch
 import torch.nn as nn
-from models.lpips import LPIPS
-from models.nets import Encoder, Decoder
 class Tokenizer(nn.Module):
     def __init__(self, vocab_size: int, embed_dim: int, encoder: Encoder, decoder: Decoder, with_lpips: bool = True) -> None:

 import torch
 import torch.nn as nn
+from .models.lpips import LPIPS
+from .models.nets import Encoder, Decoder
 class Tokenizer(nn.Module):
     def __init__(self, vocab_size: int, embed_dim: int, encoder: Encoder, decoder: Decoder, with_lpips: bool = True) -> None: