File size: 1,882 Bytes
c28358e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
#!/usr/bin/env python3
"""
Check mlx-lm version and compatibility
"""
import importlib.metadata
import logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
def check_versions():
"""Check installed package versions"""
try:
# Check mlx-lm version
mlx_lm_version = importlib.metadata.version('mlx-lm')
logger.info(f"📦 mlx-lm version: {mlx_lm_version}")
except importlib.metadata.PackageNotFoundError:
logger.error("❌ mlx-lm not installed")
return
try:
# Check MLX version
mlx_version = importlib.metadata.version('mlx')
logger.info(f"🍎 MLX version: {mlx_version}")
except importlib.metadata.PackageNotFoundError:
logger.error("❌ MLX not installed")
return
# Provide version-specific guidance
logger.info("\n💡 Version-specific notes:")
if mlx_lm_version.startswith('0.0.'):
logger.info(" - Use: generate(model, tokenizer, prompt, max_tokens)")
logger.info(" - Optional: temp, top_p parameters")
else:
logger.info(" - Check documentation for parameter names")
logger.info(" - Some parameters may have changed")
def show_available_parameters():
"""Show available parameters for generate function"""
logger.info("\n🔍 Available generate parameters (typical):")
logger.info(" - prompt: str (required)")
logger.info(" - max_tokens: int (default: 100)")
logger.info(" - temp: float (temperature, default: 0.0)")
logger.info(" - top_p: float (nucleus sampling, default: 1.0)")
logger.info(" - verbose: bool (show progress, default: False)")
logger.info(" - repetition_penalty: float (default: 1.0)")
if __name__ == "__main__":
check_versions()
show_available_parameters() |