code: def f_theta(self, x, sigma, encoder_hidden_states):
      with torch.inference_mode():
          return self.denoiser(x, sigma, encoder_hidden_states).sample
  
file_path: diffusion-for-beginners/samplers/heun.py
imports: import torch
  from math import sqrt
label_desc: Use image node_id 7 for calling the HeunSampler obj's underlying member method code with example usage: obj.f_theta(x, sigma, encoder_hidden_states) and returns: self