mergoo icon indicating copy to clipboard operation
mergoo copied to clipboard

ValueError: `rope_scaling`'s type field must be one of ['su', 'yarn'], got longrope

Open jzyee opened this issue 5 months ago • 1 comments

attempted: https://github.com/Leeroo-AI/mergoo/blob/main/notebooks/integrate_phi3_experts.ipynb

when loading the model with:

import torch from mergoo.models.modeling_phi3 import Phi3ForCausalLM

model_id = 'data/checkpoint_demo' // Define the device (use cuda:0 or another GPU if necessary) device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu') model = Phi3ForCausalLM.from_pretrained( model_id, device_map=device, torch_dtype=torch.bfloat16 )

and got the following error: ValueError: rope_scaling's type field must be one of ['su', 'yarn'], got longrope

transformers==4.44.2 torch==2.0.1

jzyee avatar Aug 31 '24 20:08 jzyee