ankahi / test_config.py
bhriguverma's picture
Add files using upload-large-folder tool
dc7e099 verified
Raw
History Blame Contribute Delete
295 Bytes
from transformers import AutoConfig
import os
MODEL_PATH = "/workspace/ankahi/punjabi_gemma/ankahi"
print(f"Testing AutoConfig from {MODEL_PATH}")
try:
config = AutoConfig.from_pretrained(MODEL_PATH)
print("Success!")
print(config)
except Exception as e:
print(f"Failed: {e}")