Browse Source

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.
pull/115/head
Doxin 6 months ago committed by GitHub
parent
commit
be595a27be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      clip_interrogator/clip_interrogator.py

2
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:

Loading…
Cancel
Save