Passing kwargs to StableDiffusion experiment

hegelai · Aug 22, 2023 · e1f55a3 · e1f55a3
1 parent 31d3440
commit e1f55a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prompttools/experiment/experiments/stablediffusion_experiment.py b/prompttools/experiment/experiments/stablediffusion_experiment.py
@@ -131,7 +131,7 @@ def run(
  )
  else:
  client = DiffusionPipeline.from_pretrained(
- model_combo["hf_model_path"], {k: call_combo[k] for k in call_combo if k != "prompt"}
+ model_combo["hf_model_path"], **{k: call_combo[k] for k in call_combo if k != "prompt"}
  )
  client.to("cuda")
  for _ in range(runs):