code: def tokenize_prompt(prompt, neg_prompt):
      prompt_ids = pipeline.prepare_inputs(prompt)
      neg_prompt_ids = pipeline.prepare_inputs(neg_prompt)
      return prompt_ids, neg_prompt_ids
  
file_path: diffusers/examples/research_projects/sdxl_flax/sdxl_single_aot.py
imports: import time
  import jax
  import jax.numpy
  import numpy
  from flax.jax_utils import replicate
  from jax import pmap
  from jax.experimental.compilation_cache import compilation_cache
  from diffusers import FlaxStableDiffusionXLPipeline
label_desc: Use image node_id 1 for calling a global function with example usage: tokenize_prompt(prompt, neg_prompt) and returns: prompt_ids,  neg_prompt_ids