import os def get_model_name(model): return os.path.basename(model.name_or_path)