search

huggingface / optimum-amd

AMD related optimizations for transformer models
https://huggingface.co/docs/optimum/amd/index
MIT License
57 stars 17 forks source link

[RyzenAI] Add default config #122

Closed mht-sharma closed 7 months ago

mht-sharma commented 7 months ago

As per title.

User can now skip providing the vaip_config as an argument. The model will use default config if none provided

This would allow to truly adapt to the existing transformers interface by just replacing class.

- model = AutoModelForImageClassification.from_pretrained(model_id)
+ model = RyzenAIModelForImageClassification.from_pretrained(model_id)