langvision 0.0.1__py3-none-any.whl → 0.1.0__py3-none-any.whl

langvision/utils/setup.py

@@ -0,0 +1,266 @@
+"""
+Setup utilities for Langvision framework initialization and validation.
+"""
+
+import torch
+import logging
+import warnings
+from typing import Optional, Dict, Any
+import sys
+import os
+from pathlib import Path
+
+
+def setup_logging(level: str = "INFO", 
+                 log_file: Optional[str] = None,
+                 format_string: Optional[str] = None) -> logging.Logger:
+    """Setup comprehensive logging for Langvision."""
+    
+    if format_string is None:
+        format_string = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    
+    # Configure root logger
+    logging.basicConfig(
+        level=getattr(logging, level.upper()),
+        format=format_string,
+        handlers=[
+            logging.StreamHandler(sys.stdout),
+        ] + ([logging.FileHandler(log_file)] if log_file else [])
+    )
+    
+    # Get langvision logger
+    logger = logging.getLogger("langvision")
+    logger.info(f"Langvision logging initialized at {level} level")
+    
+    return logger
+
+
+def validate_environment() -> Dict[str, Any]:
+    """Validate the environment for Langvision usage."""
+    
+    validation_results = {
+        "python_version": sys.version,
+        "pytorch_version": torch.__version__,
+        "cuda_available": torch.cuda.is_available(),
+        "cuda_version": None,
+        "gpu_count": 0,
+        "mps_available": False,
+        "warnings": [],
+        "errors": []
+    }
+    
+    # Check CUDA
+    if torch.cuda.is_available():
+        validation_results["cuda_version"] = torch.version.cuda
+        validation_results["gpu_count"] = torch.cuda.device_count()
+        validation_results["gpu_names"] = [
+            torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())
+        ]
+    else:
+        validation_results["warnings"].append("CUDA not available - using CPU")
+    
+    # Check MPS (Apple Silicon)
+    if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
+        validation_results["mps_available"] = True
+    
+    # Check Python version
+    if sys.version_info < (3, 8):
+        validation_results["errors"].append(
+            f"Python 3.8+ required, found {sys.version_info.major}.{sys.version_info.minor}"
+        )
+    
+    # Check PyTorch version
+    torch_version = tuple(map(int, torch.__version__.split('.')[:2]))
+    if torch_version < (1, 10):
+        validation_results["warnings"].append(
+            f"PyTorch 1.10+ recommended, found {torch.__version__}"
+        )
+    
+    return validation_results
+
+
+def setup_cuda(seed: int = 42, 
+               deterministic: bool = False,
+               benchmark: bool = True,
+               max_split_size_mb: Optional[int] = None) -> None:
+    """Setup CUDA environment for optimal performance."""
+    
+    if not torch.cuda.is_available():
+        warnings.warn("CUDA not available, skipping CUDA setup")
+        return
+    
+    # Set random seeds
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)
+    
+    # Configure CUDA settings
+    torch.backends.cudnn.deterministic = deterministic
+    torch.backends.cudnn.benchmark = benchmark
+    
+    # Memory management
+    if max_split_size_mb:
+        os.environ['PYTORCH_CUDA_ALLOC_CONF'] = f'max_split_size_mb:{max_split_size_mb}'
+    
+    # Clear cache
+    torch.cuda.empty_cache()
+    
+    logger = logging.getLogger("langvision.setup")
+    logger.info(f"CUDA setup completed with {torch.cuda.device_count()} GPUs")
+
+
+def set_seed(seed: int = 42) -> None:
+    """Set random seeds for reproducibility."""
+    import random
+    import numpy as np
+    
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed(seed)
+        torch.cuda.manual_seed_all(seed)
+    
+    # For deterministic behavior
+    torch.backends.cudnn.deterministic = True
+    torch.backends.cudnn.benchmark = False
+
+
+def check_dependencies() -> Dict[str, bool]:
+    """Check if optional dependencies are available."""
+    
+    dependencies = {}
+    
+    # Core dependencies
+    try:
+        import torch
+        dependencies["torch"] = True
+    except ImportError:
+        dependencies["torch"] = False
+    
+    try:
+        import torchvision
+        dependencies["torchvision"] = True
+    except ImportError:
+        dependencies["torchvision"] = False
+    
+    # Optional dependencies
+    try:
+        import transformers
+        dependencies["transformers"] = True
+    except ImportError:
+        dependencies["transformers"] = False
+    
+    try:
+        import wandb
+        dependencies["wandb"] = True
+    except ImportError:
+        dependencies["wandb"] = False
+    
+    try:
+        import sklearn
+        dependencies["sklearn"] = True
+    except ImportError:
+        dependencies["sklearn"] = False
+    
+    try:
+        import cv2
+        dependencies["opencv"] = True
+    except ImportError:
+        dependencies["opencv"] = False
+    
+    try:
+        import pandas
+        dependencies["pandas"] = True
+    except ImportError:
+        dependencies["pandas"] = False
+    
+    return dependencies
+
+
+def initialize_langvision(config: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+    """Initialize Langvision framework with comprehensive setup."""
+    
+    if config is None:
+        config = {}
+    
+    # Setup logging
+    log_level = config.get("log_level", "INFO")
+    log_file = config.get("log_file", None)
+    logger = setup_logging(log_level, log_file)
+    
+    logger.info("Initializing Langvision framework...")
+    
+    # Validate environment
+    validation_results = validate_environment()
+    
+    # Log validation results
+    logger.info(f"Python version: {validation_results['python_version']}")
+    logger.info(f"PyTorch version: {validation_results['pytorch_version']}")
+    
+    if validation_results["cuda_available"]:
+        logger.info(f"CUDA version: {validation_results['cuda_version']}")
+        logger.info(f"GPU count: {validation_results['gpu_count']}")
+        for i, gpu_name in enumerate(validation_results.get("gpu_names", [])):
+            logger.info(f"GPU {i}: {gpu_name}")
+    
+    if validation_results["mps_available"]:
+        logger.info("MPS (Apple Silicon) available")
+    
+    # Log warnings and errors
+    for warning in validation_results["warnings"]:
+        logger.warning(warning)
+    
+    for error in validation_results["errors"]:
+        logger.error(error)
+    
+    if validation_results["errors"]:
+        raise RuntimeError("Environment validation failed with errors")
+    
+    # Setup CUDA if requested and available
+    if config.get("setup_cuda", True) and validation_results["cuda_available"]:
+        setup_cuda(
+            seed=config.get("seed", 42),
+            deterministic=config.get("deterministic", False),
+            benchmark=config.get("benchmark", True),
+            max_split_size_mb=config.get("max_split_size_mb", None)
+        )
+    
+    # Set random seed
+    if "seed" in config:
+        set_seed(config["seed"])
+        logger.info(f"Random seed set to {config['seed']}")
+    
+    # Check dependencies
+    dependencies = check_dependencies()
+    missing_deps = [dep for dep, available in dependencies.items() if not available]
+    
+    if missing_deps:
+        logger.warning(f"Missing optional dependencies: {missing_deps}")
+        logger.info("Install with: pip install langvision[all] for full functionality")
+    
+    logger.info("Langvision framework initialization completed successfully!")
+    
+    return {
+        "validation_results": validation_results,
+        "dependencies": dependencies,
+        "config": config
+    }
+
+
+# Convenience function for quick setup
+def quick_setup(seed: int = 42, 
+               log_level: str = "INFO",
+               use_cuda: bool = True) -> None:
+    """Quick setup for Langvision with sensible defaults."""
+    
+    config = {
+        "seed": seed,
+        "log_level": log_level,
+        "setup_cuda": use_cuda,
+        "benchmark": True,
+        "deterministic": False
+    }
+    
+    initialize_langvision(config)

langvision-0.1.0.dist-info/METADATA

@@ -0,0 +1,50 @@
+Metadata-Version: 2.1
+Name: langvision
+Version: 0.1.0
+Summary: Efficient LoRA Fine-Tuning for Vision LLMs with advanced CLI and model zoo
+Home-page: https://github.com/langtrain-ai/langtrain
+Author: Pritesh Raj
+Author-email: priteshraj10@gmail.com
+Keywords: vision,transformer,lora,fine-tuning,deep-learning,computer-vision
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Requires-Dist: torch>=1.10.0
+Requires-Dist: torchvision>=0.11.0
+Requires-Dist: numpy>=1.21.0
+Requires-Dist: pillow>=8.3.0
+Requires-Dist: tqdm>=4.62.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: tensorboard>=2.9.0
+Requires-Dist: wandb>=0.13.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.8.0; extra == "dev"
+Requires-Dist: black>=22.0.0; extra == "dev"
+Requires-Dist: isort>=5.10.0; extra == "dev"
+Requires-Dist: flake8>=5.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Requires-Dist: pre-commit>=2.20.0; extra == "dev"
+Requires-Dist: bandit>=1.7.0; extra == "dev"
+Provides-Extra: docs
+Requires-Dist: sphinx>=5.0.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme>=1.0.0; extra == "docs"
+Requires-Dist: myst-parser>=0.18.0; extra == "docs"
+Requires-Dist: sphinx-autodoc-typehints>=1.19.0; extra == "docs"
+Provides-Extra: examples
+Requires-Dist: jupyter>=1.0.0; extra == "examples"
+Requires-Dist: ipywidgets>=7.6.0; extra == "examples"
+Requires-Dist: tensorboard>=2.9.0; extra == "examples"
+Requires-Dist: wandb>=0.13.0; extra == "examples"
+

langvision-0.1.0.dist-info/RECORD

@@ -0,0 +1,61 @@
+langvision/__init__.py,sha256=1K-pmG7huJTVWdGXiWQmmvaI2QYJoFVty2X-5kkebu8,2758
+langvision/example.py,sha256=KpPS_3hV4eFBysTRH4qcVQffVlkcwcWjDhhafbcXku0,85
+langvision/model_zoo.py,sha256=S3_rtAQV_eswIyJICfiwoBw9jQqlXy9fu0irlM8GETs,5491
+langvision/agents/__init__.py,sha256=8whI81DC6F9ZoV24_GOzD8LwYKcUXIc53TgWQILM6ng,141
+langvision/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langvision/callbacks/base.py,sha256=hnShvmpOtv3fH9LAPzklOOx1Xi1v32awb9i8F0LZtCM,6794
+langvision/callbacks/early_stopping.py,sha256=LKWO5TiQ05WCNabO1utm9tdOfBgp9lwRmIGmXO73Gaw,543
+langvision/callbacks/logging.py,sha256=Wk882TpUa_CyhbGKgp8IZyOUYuQPnO5l-uBACWdxFKs,404
+langvision/callbacks/registry.py,sha256=70Zq-hYWBuqxxSKEOUbKEBEdwdHY3qJc8GLpz7am0IM,191
+langvision/cli/__init__.py,sha256=WEFs8f61xkxialeYOW2A67utuT6JSRRoTcgGcFuWoSc,3381
+langvision/cli/complete_cli.py,sha256=Dqv-eAhMTNyq8YWsofAuPhHX6NkYfs8hB5wZlYUGm7o,13870
+langvision/cli/config.py,sha256=nuOgEhBMdetDmaF9-KI-58LBUsloCCd2JYKrnEiemSg,13181
+langvision/cli/evaluate.py,sha256=a4KsI61BGZ3X-16ruGQRqWMANV-cVYuecsrzPcNdpmY,7593
+langvision/cli/export.py,sha256=ufkB7sa1vNTwW0WT-PsxP3Jsvpf87wgkYIFZSatkSf0,6781
+langvision/cli/finetune.py,sha256=66f88ViTImq7mC6pqq39_76WRN7OnBs4y4J0zYEU2Zk,14668
+langvision/cli/model_zoo.py,sha256=lbuiYiLu39a3MWGB24CpsY8684rv4wCcpfknr9YogeE,6369
+langvision/cli/train.py,sha256=RFPV0GsH9cexfUVM1Sbk_nqcLOK9QD1cswWL9AkLmKY,5301
+langvision/cli/utils.py,sha256=-Ymmr_-x2U-za13KF5kfmdnpxCs8zXiyNh95y8qjQBs,7509
+langvision/components/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
+langvision/components/attention.py,sha256=NVWIEkZ8pIi8K3iS5KQpKC3hJgMl7UozIFf7vkpx9Ro,2163
+langvision/components/mlp.py,sha256=t9ELFMGLhG8zTICG5xRmrYm7b-IcszRUNGIBSwBVCas,246
+langvision/components/patch_embedding.py,sha256=C_HFjD4rixhxoxg2QhRjY-j_iwd0_eCnPdveVBos3D8,574
+langvision/concepts/__init__.py,sha256=_kYp1XEiLJIQQz0s_lM11K8r-qGc4vdJtpUNntt57RA,201
+langvision/concepts/ccot.py,sha256=yIgvGOdezW8ZtmxodLvMUtzF4JzxuZw-6lCK16UHXY0,1149
+langvision/concepts/cot.py,sha256=KwUpP3JbXu6GfPfl_jnZC1siXs3VgETCnmdXW94jMug,941
+langvision/concepts/dpo.py,sha256=o6mCuV-axjA61j_fslvHGcJPgdupX5j-XPC0lYvXgOU,1428
+langvision/concepts/grpo.py,sha256=QzAOaOzMcIhJTWqCjMOIrDYV3hoa6R3BRe9t1wV-hCc,810
+langvision/concepts/lime.py,sha256=NUO8-rC7AtjMLFKNOc-a87uA_4zPSXKfBILNMdwBF-g,1500
+langvision/concepts/ppo.py,sha256=MpnuDsxbTM2iPnwKRkM3qtxXNxrZfzaM1tudLE-Wc8E,1855
+langvision/concepts/rlhf.py,sha256=0v6tiqi8LXwgYDKcOYzzgWkq5eaAFSSE_yJbrUs-JXs,1594
+langvision/concepts/rlvr.py,sha256=wfhODtH_rw3dIiEe4ZgDaNVGPLmKnyQg01aK14Uc9vU,765
+langvision/concepts/shap.py,sha256=c1NzDO5QoKn_rnG2uW4RPrtnObWaBIl6INWwn6gIVwM,1295
+langvision/config/__init__.py,sha256=1V4Mn4_qbz0xmPYcboNjorImNkqYyRlL5r0amFdVu98,312
+langvision/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langvision/data/datasets.py,sha256=B27d0BX87RR1kNlwuT1QUszSP-b3zbCKQVR2lebIs1A,739
+langvision/data/enhanced_datasets.py,sha256=G7S_o74tDVRXk5vtSg4vchZiGd-Drt4uvwUxaonylNU,22023
+langvision/filesystem/__init__.py,sha256=XvdZcHNCAk3mdLesBIdUEvp9Of9wVlYKqAfDGDmqbbA,318
+langvision/llm/__init__.py,sha256=KFZ0plTNQU72wS3cLUxEbz15HtcOhBJrdt4rZUnOyvE,85
+langvision/memory/__init__.py,sha256=j3hnLDohdr_eVOHCgvPU7XlG3taBbC1u-YhR_Kqd5xw,487
+langvision/models/__init__.py,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
+langvision/models/lora.py,sha256=SfY7csz8kUQWI26ZmeQTO_twffxeT_Wpon9S0oO2fZs,7582
+langvision/models/multimodal.py,sha256=VYXM3jHl3Yr5_cdOYyn0e4G7mQ2NEi3LELwtgdvXeP0,11953
+langvision/models/resnet.py,sha256=0EObpib94xywm1-5wxSUVwAg6uGxoZZaOvBIX8eaVVI,11733
+langvision/models/vision_transformer.py,sha256=oA01la5qvRn8i4lGaQ7da71_ExHmNtGq2mrqUByIpNs,1244
+langvision/sync/__init__.py,sha256=MdC6UeCjrrsrmaOy1qNK0hA9ijd1CJvWMkVHo7B7gwU,373
+langvision/telemetry/__init__.py,sha256=DMz1ACz7KtSOr2uqtKfEQ_yk_PrvdlH2zHeMH-NdHiA,210
+langvision/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+langvision/training/advanced_trainer.py,sha256=OrfOjDomM5XZFAVfrXBTUrTGzLjQdxwnqq4KTkiZrnM,18438
+langvision/training/trainer.py,sha256=xqtyUkUMiHZLJvGCc8iqfODKmuXxsadPxSsRA_engTE,5635
+langvision/utils/__init__.py,sha256=xwagF_TAjkaFPDMv67lqOZblpkiWBrM5lwNYpntvFds,828
+langvision/utils/config.py,sha256=KALv8nP8WN31Q9Uij2ihF6tgp4z8u9MxtU42hIarEoA,5453
+langvision/utils/cuda.py,sha256=rpp0j-tvwGK6uWVe7fFTOS1BdZcsfRkw2AS5B08L-R0,1372
+langvision/utils/data.py,sha256=95U8Z_R2slwpaBVFrqyvcBA6kYINcEuzp1r5djvwQYg,277
+langvision/utils/device.py,sha256=TLVwXHFcildDyKh1Q7zczIyU702UJj7OHpsGhjRwLjk,552
+langvision/utils/metrics.py,sha256=gcPKrAamz2YKGKk07bdT0DxpgnQc3DHcccnK5CB_dgw,17272
+langvision/utils/setup.py,sha256=tHv9_G8q0C0VAZ5UqN8HIIz4LX5QVSqrd1PKqYFp9P0,8046
+langvision-0.1.0.dist-info/METADATA,sha256=qyiyvK3EO6CphApDm5y53OpY_klOBjDPPhznOlq8Pr0,2129
+langvision-0.1.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+langvision-0.1.0.dist-info/entry_points.txt,sha256=KAVJFWwV8QDww6yOit8YrkhJDujHd8RKFbuKcRpdajg,51
+langvision-0.1.0.dist-info/top_level.txt,sha256=l3g51kgU-5cacP4nhASSfYADOCAwLC4UzlgVwuoJCgc,11
+langvision-0.1.0.dist-info/RECORD,,

{langvision-0.0.1.dist-info → langvision-0.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

langvision-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+langvision = langvision.cli:main