Update llamaPmi.py