Returned original imports.

src/open_clip/factory.py

@@ -9,18 +9,16 @@
 
 import torch
 
+from .constants import OPENAI_DATASET_MEAN, OPENAI_DATASET_STD
+from .model import CLIP, CustomTextCLIP, convert_weights_to_lp, convert_to_custom_text_state_dict,\
+    resize_pos_embed, get_cast_dtype, resize_text_pos_embed, set_model_preprocess_cfg
 from .coca_model import CoCa
-from .loss import ClipLoss, CoCaLoss, DistillClipLoss, SigLipLoss
-from .model import (CLIP, CustomTextCLIP, convert_to_custom_text_state_dict,
-                    convert_weights_to_lp, get_cast_dtype, resize_pos_embed,
-                    resize_text_pos_embed, set_model_preprocess_cfg)
+from .loss import ClipLoss, DistillClipLoss, CoCaLoss, SigLipLoss
 from .openai import load_openai_model
-from .pretrained import (download_pretrained, download_pretrained_from_hf,
-                         get_pretrained_cfg, list_pretrained_tags_by_model)
-from .tokenizer import (DEFAULT_CONTEXT_LENGTH, HFTokenizer, NLLBTokenizer,
-                        SimpleTokenizer)
-from .transform import (AugmentationCfg, PreprocessCfg, image_transform_v2,
-                        merge_preprocess_dict, merge_preprocess_kwargs)
+from .pretrained import is_pretrained_cfg, get_pretrained_cfg, download_pretrained,\
+    list_pretrained_tags_by_model, download_pretrained_from_hf
+from .transform import image_transform_v2, AugmentationCfg, PreprocessCfg, merge_preprocess_dict, merge_preprocess_kwargs
+from .tokenizer import HFTokenizer, SimpleTokenizer, DEFAULT_CONTEXT_LENGTH
 
 HF_HUB_PREFIX = 'hf-hub:'
 _MODEL_CONFIG_PATHS = [Path(__file__).parent / f"model_configs/"]