Covert chemaleon weights to hf, ImportError

ChinChyi commented 2 weeks ago

System Info

ImportError: cannot import name 'ChameleonForCausalLM' from 'transformers'  (/mnt/petrelfs/xxxx/miniconda3/envs/lumina_mgpt/lib/python3.10/site-packages/transformers/__init__.py)

Information

[X] The official example scripts
[ ] My own modified scripts

Tasks

[ ] An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
[X] My own task or dataset (give details below)

Reproduction

python src/transformers/models/chameleon/convert_chameleon_weights_to_hf.py --input_dir /mnt/petrelfs/q-pth --model_size 7B --output_dir /mnt/petrelfs

Expected behavior

Hope this is fixed

Checklist

[X] I have read the migration guide in the readme. (pytorch-transformers; pytorch-pretrained-bert)
[X] I checked if a related official extension example runs on my machine.

ArthurZucker commented 2 weeks ago

Hey @ChinChyi, you should make sure that transformers is installed using pip install -e .

ChinChyi commented 2 weeks ago

Hey @ChinChyi, you should make sure that transformers is installed using pip install -e .

@ArthurZucker I've installed transformers with this and I'm still getting this error

ArthurZucker commented 2 weeks ago

Ah you are completely right! The class is ChameleonForConditionalGeneration ! Would you like to open a PR to fix the class in the conversion script? 🤗

huggingface / transformers