Commit e2744fb4 authored by gd1551's avatar gd1551 Committed by GitHub

use prompt mixing when advanced is off

parent da5dc4d0
...@@ -120,7 +120,7 @@ class StableDiffusionModel(nn.Module): ...@@ -120,7 +120,7 @@ class StableDiffusionModel(nn.Module):
], device=self.device) ], device=self.device)
prompt = [request.prompt] * request.n_samples prompt = [request.prompt] * request.n_samples
prompt_condition = self.model.get_learned_conditioning(prompt) prompt_condition = prompt_mixing(self.model, prompt[0], request.n_samples)
uc = None uc = None
if request.scale != 1.0: if request.scale != 1.0:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment