mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
Use pre-mode quant on custom transformer files
This commit is contained in:
@@ -42,8 +42,8 @@ def load_transformer(repo_id, cls_name, load_config={}, subfolder="transformer",
|
||||
local_file,
|
||||
cache_dir=shared.opts.hfcache_dir,
|
||||
**load_args,
|
||||
**quant_args,
|
||||
)
|
||||
transformer = model_quant.do_post_load_quant(transformer, allow=quant_type is not None)
|
||||
else:
|
||||
shared.log.debug(f'Load model: transformer="{repo_id}" cls={cls_name.__name__} quant="{quant_type}" args={load_args}')
|
||||
if dtype is not None:
|
||||
|
||||
Reference in New Issue
Block a user