code: def add_noise_common(
      state: CommonSchedulerState,
      original_samples: jnp.ndarray,
      noise: jnp.ndarray,
      timesteps: jnp.ndarray,
  ):
      sqrt_alpha_prod, sqrt_one_minus_alpha_prod = get_sqrt_alpha_prod(
          state, original_samples, noise, timesteps
      )
      noisy_samples = (
          sqrt_alpha_prod * original_samples
          + sqrt_one_minus_alpha_prod * noise
      )
      return noisy_samples
  
imports: import importlib
  import math
  import os
  from dataclasses import dataclass
  from enum import Enum
  from typing import Optional, Tuple, Union
  import flax
  import jax.numpy
  from ..utils import BaseOutput, PushToHubMixin
label_desc: Use image node_id 4 for calling a global function with example usage: add_noise_common(state, original_samples, noise, timesteps) and returns: noisy_samples