Revert "much stricter ruff linting"

This reverts commit 310dbf1574.

scripts/pulid/eva_clip/factory.py

@@ -18,7 +18,7 @@ from .tokenizer import HFTokenizer, tokenize
 from .utils import resize_clip_pos_embed, resize_evaclip_pos_embed, resize_visual_pos_embed, resize_eva_pos_embed
 
 
-_MODEL_CONFIG_PATHS = [Path(__file__).parent / "model_configs/"]
+_MODEL_CONFIG_PATHS = [Path(__file__).parent / f"model_configs/"]
 _MODEL_CONFIGS = {}  # directory (model_name: config) of model architecture configs
 
 

scripts/pulid/eva_clip/hf_model.py

@@ -14,7 +14,7 @@ try:
     from transformers import AutoModel, AutoModelForMaskedLM, AutoTokenizer, AutoConfig, PretrainedConfig
     from transformers.modeling_outputs import BaseModelOutput, BaseModelOutputWithPooling, \
         BaseModelOutputWithPoolingAndCrossAttentions
-except ImportError:
+except ImportError as e:
     transformers = None
 
 

scripts/pulid/eva_clip/model.py

@@ -385,7 +385,7 @@ def build_model_from_openai_state_dict(
     vocab_size = state_dict["token_embedding.weight"].shape[0]
     transformer_width = state_dict["ln_final.weight"].shape[0]
     transformer_heads = transformer_width // 64
-    transformer_layers = len(set(k.split(".")[2] for k in state_dict if k.startswith("transformer.resblocks")))
+    transformer_layers = len(set(k.split(".")[2] for k in state_dict if k.startswith(f"transformer.resblocks")))
 
     vision_cfg = CLIPVisionCfg(
         layers=vision_layers,

scripts/pulid/eva_clip/timm_model.py

@@ -110,7 +110,7 @@ class TimmModel(nn.Module):
     def set_grad_checkpointing(self, enable=True):
         try:
             self.trunk.set_grad_checkpointing(enable)
-        except Exception:
+        except Exception as e:
             logging.warning('grad checkpointing not supported for this timm image tower, continuing without...')
 
     def forward(self, x):

scripts/pulid/eva_clip/tokenizer.py

@@ -12,6 +12,7 @@ import regex as re
 import torch
 
 # https://stackoverflow.com/q/62691279
+import os
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 

scripts/pulid/eva_clip/utils.py

@@ -149,7 +149,7 @@ def resize_rel_pos_embed(state_dict, model, interpolation: str = 'bicubic', seq_
             dst_num_pos, _ = model.visual.state_dict()[key].size()
             dst_patch_shape = model.visual.patch_embed.patch_shape
             if dst_patch_shape[0] != dst_patch_shape[1]:
-                raise NotImplementedError
+                raise NotImplementedError()
             num_extra_tokens = dst_num_pos - (dst_patch_shape[0] * 2 - 1) * (dst_patch_shape[1] * 2 - 1)
             src_size = int((src_num_pos - num_extra_tokens) ** 0.5)
             dst_size = int((dst_num_pos - num_extra_tokens) ** 0.5)

scripts/pulid/pulid_flux.py

@@ -6,7 +6,7 @@ from modules.logger import log
 
 
 def apply_flux(pipe: FluxPipeline):
-    if not hasattr(pipe, 'transformer') or 'Nunchaku' not in pipe.transformer.__class__.__name__:
+    if not hasattr(pipe, 'transformer') or not 'Nunchaku' in pipe.transformer.__class__.__name__:
         log.error('PuLID: flux support requires nunchaku')
         return pipe