Get-Things-Done-with-Prompt-Engineering-and-LangChain icon indicating copy to clipboard operation
Get-Things-Done-with-Prompt-Engineering-and-LangChain copied to clipboard

I need to load my custom model from pre trained model

Open shaileshp367 opened this issue 2 years ago • 0 comments

Hello

I need to load my custom model from pre trained model

Here is the code but he giving error like 'HFValidationError: Repo id must use alphanumeric chars or '-', '_', '.', '--' and '..' are forbidden, '-' and '.' cannot start or end the name, max length is'

Please Help

model = "shaileshp/trained-test-model-1-merged"

dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] == 8 else torch.float16

model = AutoModelForCausalLM.from_pretrained( model, return_dict=True, device_map="auto", load_in_8bit=True, torch_dtype=dtype, #trust_remote_code=True, )

shaileshp367 avatar Jun 27 '23 11:06 shaileshp367