def _lora_weights_candidates()

in docker_images/diffusers/app/lora.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

    def _lora_weights_candidates(model_data):
        candidate = None
        for file in model_data.siblings:
            rfilename = str(file.rfilename)
            if rfilename.endswith("pytorch_lora_weights.bin"):
                candidate = rfilename
            elif rfilename.endswith("pytorch_lora_weights.safetensors"):
                candidate = rfilename
                break
        return candidate