Unable to load model

#16
by xDynames - opened

When trying to call torch.load on training_args.bin I get an AttributeError: Can't get attribute 'TrainingArguments' on <module '__main__' (built-in)> error.

This prevents it being used with https://github.com/huggingface/text-generation-inference/ even though Llama models are supported.

Any help getting this working would be greatly appreciated

Sign up or log in to comment