--text-encoders help fix

Teriks · Jun 26, 2024 · 496a444 · 496a444
1 parent d4e55cc
commit 496a444
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/dgenerate/pipelinewrapper/pipelines.py b/dgenerate/pipelinewrapper/pipelines.py
@@ -968,10 +968,10 @@ def _text_encoder_not_null(uri):
 
 def _torch_args_hasher(args):
     def text_encoder_uri_parse(uri):
-        if uri is None or uri == '+':
+        if uri is None or uri.strip() == '+':
             return None
 
-        if uri == 'help':
+        if uri.strip() == 'help':
             return 'help'
 
         return _uris.TorchTextEncoderUri.parse(uri)
@@ -1645,10 +1645,10 @@ def __call__(self) -> FlaxPipelineCreationResult:
 def _flax_args_hasher(args):
 
     def text_encoder_uri_parse(uri):
-        if uri is None or uri == '+':
+        if uri is None or uri.strip() == '+':
             return None
 
-        if uri == 'help':
+        if uri.strip() == 'help':
             return 'help'
 
         return _uris.FlaxTextEncoderUri.parse(uri)