code: def _p_generate(
      pipe,
      prompt_ids,
      mask,
      masked_image,
      params,
      prng_seed,
      num_inference_steps,
      height,
      width,
      guidance_scale,
      latents,
      neg_prompt_ids,
  ):
      return pipe._generate(
          prompt_ids,
          mask,
          masked_image,
          params,
          prng_seed,
          num_inference_steps,
          height,
          width,
          guidance_scale,
          latents,
          neg_prompt_ids,
      )
  
imports: import warnings
  from functools import partial
  from typing import Dict, List, Optional, Union
  import jax
  import jax.numpy
  import numpy
  from flax.core.frozen_dict import FrozenDict
  from flax.jax_utils import unreplicate
  from flax.training.common_utils import shard
  from packaging import version
  from PIL import Image
  from transformers import CLIPImageProcessor, CLIPTokenizer, FlaxCLIPTextModel
  from ...models import FlaxAutoencoderKL, FlaxUNet2DConditionModel
  from ...schedulers import FlaxDDIMScheduler, FlaxDPMSolverMultistepScheduler, FlaxLMSDiscreteScheduler, FlaxPNDMScheduler
  from ...utils import PIL_INTERPOLATION, deprecate, logging, replace_example_docstring
  from ..pipeline_flax_utils import FlaxDiffusionPipeline
  from .pipeline_output import FlaxStableDiffusionPipelineOutput
  from .safety_checker_flax import FlaxStableDiffusionSafetyChecker
label_desc: Use image node_id 1 for calling a global function with example usage: _p_generate(pipe, prompt_ids, mask, masked_image, params, prng_seed, num_inference_steps, height, width, guidance_scale, latents, neg_prompt_ids) and returns: pipe