From be595a27be55cceecb860487c0e400374f18a283 Mon Sep 17 00:00:00 2001 From: Doxin Date: Wed, 15 May 2024 15:32:19 +0200 Subject: [PATCH] Split model name correctly `String.split(..., 2)` will split the string at most 2 times, resulting in a 3-tuple. Presumably the intent here was to have a 2-tuple at most, meaning the string gets split at most once. --- clip_interrogator/clip_interrogator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clip_interrogator/clip_interrogator.py b/clip_interrogator/clip_interrogator.py index fd2fe68..b45c94a 100644 --- a/clip_interrogator/clip_interrogator.py +++ b/clip_interrogator/clip_interrogator.py @@ -96,7 +96,7 @@ class Interrogator(): start_time = time.time() config = self.config - clip_model_name, clip_model_pretrained_name = config.clip_model_name.split('/', 2) + clip_model_name, clip_model_pretrained_name = config.clip_model_name.split('/', 1) if config.clip_model is None: if not config.quiet: