Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-21 21:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add threshold for clip image encoder conversion

Browse source
This commit is contained in:
Cédric Deltheil 2023-09-07 14:46:15 +02:00 committed by Cédric Deltheil
parent c6fadd1c81
commit 946e7c2974
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/conversion/convert_transformers_clip_image_model.py
View file

@ -15,6 +15,7 @@ class Args(argparse.Namespace):
output_path: str | None
half: bool
verbose: bool
threshold: float
def setup_converter(args: Args) -> ModelConverter:
@ -79,7 +80,7 @@ def setup_converter(args: Args) -> ModelConverter:
# Ad hoc post-conversion steps
class_embedding.parameter = torch.nn.Parameter(source.vision_model.embeddings.class_embedding.clone()) # type: ignore
assert converter.compare_models((x,), threshold=1e-2)
assert converter.compare_models((x,), threshold=args.threshold)
return converter
@ -122,6 +123,7 @@ def main() -> None:
default=False,
help="Prints additional information during conversion. Default: False",
)
parser.add_argument("--threshold", type=float, default=1e-2, help="Threshold for model comparison. Default: 1e-2")
args = parser.parse_args(namespace=Args())
if args.output_path is None:
args.output_path = f"{Path(args.source_path).stem}-{args.subfolder}.safetensors"