()
| 58 | |
| 59 | @cache_model |
| 60 | def load_image_encoder(): |
| 61 | image_encoder = CLIPVisionModelWithProjection.from_pretrained( |
| 62 | "h94/IP-Adapter", |
| 63 | subfolder="models/image_encoder", |
| 64 | torch_dtype=torch.float16, |
| 65 | ) |
| 66 | return image_encoder |
| 67 | |
| 68 | def load_common_sd15_pipe(base_model=DEFAULT_BASE_MODEL, device="auto", controlnet=None, ip_adapter=False, plus_model=True, torch_dtype=torch.float16, model_cpu_offload_seq=None, enable_sequential_cpu_offload=False, vae_slicing=False, pipeline_class=None, **kwargs): |
| 69 | model_kwargs = dict( |
no outgoing calls
no test coverage detected