6Morpheus6
/

IndicF5

Model card Files Files and versions

svp19 commited on Mar 11, 2025

Commit

1031fb0

·

1 Parent(s): d5f414f

Update model.py

Files changed (1) hide show

model.py +2 -2

model.py CHANGED Viewed

@@ -47,8 +47,8 @@ class INF5Model(PreTrainedModel):
         # print(f"Loading model weights from {safetensors_path} (safetensors)...")
         # state_dict = load_file(safetensors_path, device=str(device))
-        # # Download vocab.txt from HF Hub
-        # vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")
         self.ema_model = load_model(
                 DiT,

         # print(f"Loading model weights from {safetensors_path} (safetensors)...")
         # state_dict = load_file(safetensors_path, device=str(device))
+        # Download vocab.txt from HF Hub
+        vocab_path = hf_hub_download(config.name_or_path, filename="checkpoints/vocab.txt")
         self.ema_model = load_model(
                 DiT,