Skip to content

Commit abace05

Browse files
sayakpaulsayakpaul
committed
up
1 parent 2d4f144 commit abace05

File tree

49 files changed

+72
-73
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

49 files changed

+72
-73
lines changed

‎examples/community/adaptive_mask_inpainting.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,6 @@ def enable_model_cpu_offload(self, gpu_id=0):
489489
# We'll offload the last model manually.
490490
self.final_offload_hook = hook
491491

492-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
493492
def _encode_prompt(
494493
self,
495494
prompt,
@@ -651,7 +650,7 @@ def _encode_prompt(
651650

652651
return prompt_embeds
653652

654-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
653+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
655654
def run_safety_checker(self, image, device, dtype):
656655
if self.safety_checker is None:
657656
has_nsfw_concept = None

‎examples/community/fresco_v2v.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1380,7 +1380,7 @@ def __init__(
13801380
flow_model.eval()
13811381
self.flow_model = flow_model
13821382

1383-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
1383+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
13841384
def _encode_prompt(
13851385
self,
13861386
prompt,
@@ -1672,7 +1672,7 @@ def prepare_ip_adapter_image_embeds(
16721672

16731673
return image_embeds
16741674

1675-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
1675+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
16761676
def run_safety_checker(self, image, device, dtype):
16771677
if self.safety_checker is None:
16781678
has_nsfw_concept = None

‎examples/community/latent_consistency_interpolate.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -459,7 +459,7 @@ def encode_prompt(
459459

460460
return prompt_embeds, negative_prompt_embeds
461461

462-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
462+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
463463
def run_safety_checker(self, image, device, dtype):
464464
if self.safety_checker is None:
465465
has_nsfw_concept = None

‎examples/community/llm_grounded_diffusion.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1195,7 +1195,7 @@ def latent_lmd_guidance(
11951195
# Below are methods copied from StableDiffusionPipeline
11961196
# The design choice of not inheriting from StableDiffusionPipeline is discussed here: https://github.com/huggingface/diffusers/pull/5993#issuecomment-1834258517
11971197

1198-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
1198+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
11991199
def _encode_prompt(
12001200
self,
12011201
prompt,
@@ -1426,7 +1426,7 @@ def encode_image(self, image, device, num_images_per_prompt):
14261426
uncond_image_embeds = torch.zeros_like(image_embeds)
14271427
return image_embeds, uncond_image_embeds
14281428

1429-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
1429+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
14301430
def run_safety_checker(self, image, device, dtype):
14311431
if self.safety_checker is None:
14321432
has_nsfw_concept = None

‎examples/community/pipeline_fabric.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def __init__(
186186
self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1) if getattr(self, "vae", None) else 8
187187
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
188188

189-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
189+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
190190
def _encode_prompt(
191191
self,
192192
prompt,

‎examples/community/pipeline_hunyuandit_differential_img2img.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -512,7 +512,7 @@ def encode_prompt(
512512
negative_prompt_attention_mask,
513513
)
514514

515-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
515+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
516516
def run_safety_checker(self, image, device, dtype):
517517
if self.safety_checker is None:
518518
has_nsfw_concept = None

‎examples/community/pipeline_prompt2prompt.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -213,7 +213,7 @@ def __init__(
213213
self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
214214
self.register_to_config(requires_safety_checker=requires_safety_checker)
215215

216-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
216+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
217217
def _encode_prompt(
218218
self,
219219
prompt,
@@ -430,7 +430,7 @@ def encode_prompt(
430430

431431
return prompt_embeds, negative_prompt_embeds
432432

433-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
433+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
434434
def run_safety_checker(self, image, device, dtype):
435435
if self.safety_checker is None:
436436
has_nsfw_concept = None

‎examples/community/stable_diffusion_reference.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -358,7 +358,7 @@ def check_inputs(
358358
"Provide either `ip_adapter_image` or `ip_adapter_image_embeds`. Cannot leave both `ip_adapter_image` and `ip_adapter_image_embeds` defined."
359359
)
360360

361-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
361+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
362362
def _encode_prompt(
363363
self,
364364
prompt: Union[str, List[str]],
@@ -789,7 +789,7 @@ def prepare_ref_latents(
789789
ref_image_latents = ref_image_latents.to(device=device, dtype=dtype)
790790
return ref_image_latents
791791

792-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
792+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
793793
def run_safety_checker(
794794
self, image: Union[torch.Tensor, PIL.Image.Image], device: torch.device, dtype: torch.dtype
795795
) -> Tuple[Union[torch.Tensor, PIL.Image.Image], Optional[bool]]:

‎examples/community/stable_diffusion_repaint.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def __init__(
281281
self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1) if getattr(self, "vae", None) else 8
282282
self.register_to_config(requires_safety_checker=requires_safety_checker)
283283

284-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline._encode_prompt
284+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin._encode_prompt
285285
def _encode_prompt(
286286
self,
287287
prompt,
@@ -427,7 +427,7 @@ def _encode_prompt(
427427

428428
return prompt_embeds
429429

430-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
430+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
431431
def run_safety_checker(self, image, device, dtype):
432432
if self.safety_checker is not None:
433433
safety_checker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(device)

‎examples/community/stable_diffusion_tensorrt_img2img.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -832,7 +832,7 @@ def __loadModels(self):
832832
if "vae_encoder" in self.stages:
833833
self.models["vae_encoder"] = make_VAEEncoder(self.vae, **models_args)
834834

835-
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.run_safety_checker
835+
# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_utils.StableDiffusionMixin.run_safety_checker
836836
def run_safety_checker(
837837
self, image: Union[torch.Tensor, PIL.Image.Image], device: torch.device, dtype: torch.dtype
838838
) -> Tuple[Union[torch.Tensor, PIL.Image.Image], Optional[bool]]:

0 commit comments

Comments
 (0)