darkspark

Trace your own

darkspark can trace any PyTorch model.

  1. 1Install with pip install darkspark
  2. 2Wrap your forward pass in darkspark.Tracer()
    examples
    1import darkspark
    2from diffusers import Flux2KleinPipeline
    3​
    4pipe = Flux2KleinPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-4B")
    5​
    6with darkspark.Tracer():  # add this line
    7    pipe("a cat holding a sign", num_inference_steps=4)