File size: 668 Bytes
970c0d3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import transformers
from transformers import AutoConfig, AutoModelForCausalLM
model_id = "meta-models/Muse-Glimmer-30B"
cfg = AutoConfig.from_pretrained(model_id, trust_remote_code=True)
print("transformers:", transformers.__version__)
print("config:", type(cfg).__name__)
print("model_type:", cfg.model_type)
print("architectures:", cfg.architectures)
print("AutoModelForCausalLM supports config:",
type(cfg) in AutoModelForCausalLM._model_mapping)
image_text = getattr(transformers, "AutoModelForImageTextToText", None)
if image_text is not None:
print("AutoModelForImageTextToText supports config:",
type(cfg) in image_text._model_mapping)
|