I need to load my custom model from pre trained model
shaileshp367 opened this issue · 0 comments
shaileshp367 commented
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,
)