mosaicml

/

mpt-7b-storywriter

like 823

Follow

Mosaic ML, Inc. 353

Please check and advise me how to solve error AttributeError: module 'torch' has no attribute 'bfloat16'.

import transformers

model = transformers.AutoModelForCausalLM.from_pretrained(
  'mosaicml/mpt-7b-instruct',
  trust_remote_code=True,
)

for your reference, i have torchaudio-2.0.1 torchvision-0.15.1 torch-2.0.0 transformers-4.28.1 (apple silicon)