vikhyatk

/

moondream2

like 977

Hi,

I have been able to finetune this model and subsequently save the finetuned model with:
moondream.save_pretrained("checkpoints/moondream-ft")

But, when I later tried to load the finetuned model for evaluation with:
config = AutoConfig.from_pretrained("checkpoints/moondream-ft", trust_remote_code=True)
moondream = AutoModelForCausalLM.from_pretrained("checkpoints/moondream-ft", config=config, trust_remote_code=True, device_map={"": DEVICE})

I got the following error:
AttributeError: module 'transformers_modules.vikhyatk.moondream2.fb2293ab2450beb1dae536c056f5976becD58e4c.moondream' has no attribute 'Moondream'

I am not sure how to go about loading and using the finetuned model from here.
Any ideas?