in optimum/amd/brevitas/quantizer.py [0:0]
def from_pretrained(
cls,
model_name_or_path: str,
subfolder: str = "",
revision: Optional[str] = None,
cache_dir: Optional[str] = None,
trust_remote_code: bool = False,
force_download: bool = False,
local_files_only: bool = False,
use_auth_token: Optional[Union[bool, str]] = None,
device_map: Optional[Union[Dict, str, torch.device]] = None,
**model_kwargs,