Fix HF Sequence type alias in Stage 2 loader

code/train_production.py

@@ -28,7 +28,7 @@ import torch.nn.functional as F
 from datasets import Dataset as HFDataset
 from datasets import concatenate_datasets, load_dataset
 from datasets import Features, Image as HFImage
-from datasets.features import Sequence
+from datasets.features import Sequence as HFSequence
 from huggingface_hub import HfApi, snapshot_download
 from PIL import Image
 from torch.optim import AdamW
@@ -542,8 +542,8 @@ def build_llava_records(max_samples: Optional[int]) -> HFDataset:
 def disable_image_decoding(feature: object) -> object:
     if isinstance(feature, HFImage):
         return HFImage(decode=False)
-    if isinstance(feature, Sequence):
-        return Sequence(feature=disable_image_decoding(feature.feature), length=feature.length)
+    if isinstance(feature, HFSequence):
+        return HFSequence(feature=disable_image_decoding(feature.feature), length=feature.length)
     if isinstance(feature, Features):
         return Features({key: disable_image_decoding(value) for key, value in feature.items()})
     if isinstance(feature, dict):