6Morpheus6
/

IndicF5

Model card Files Files and versions

svp19 commited on Mar 11, 2025

Commit

cd61fb5

·

1 Parent(s): 90dfc08

Update model.py

Files changed (1) hide show

model.py +0 -3

model.py CHANGED Viewed

@@ -50,9 +50,6 @@ class INF5Model(PreTrainedModel):
         # Download vocab.txt from HF Hub
         vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")
-        self.ema_model.to(self.device)
-        self.vocoder.to(self.device)
         self.ema_model = torch.compile(load_model(
                 DiT,
                 dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),

         # Download vocab.txt from HF Hub
         vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")
         self.ema_model = torch.compile(load_model(
                 DiT,
                 dict(dim=1024, depth=22, heads=16, ff_mult=2, text_dim=512, conv_layers=4),