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

fix prior check

parent f5bcaf6a
...@@ -240,7 +240,7 @@ class StableDiffusionModel(nn.Module): ...@@ -240,7 +240,7 @@ class StableDiffusionModel(nn.Module):
prompt = [request.prompt] * request.n_samples prompt = [request.prompt] * request.n_samples
prompt_condition = prompt_mixing(self.model, prompt[0], request.n_samples) prompt_condition = prompt_mixing(self.model, prompt[0], request.n_samples)
if self.prior and request.mitigate: if 'prior' in self and request.mitigate:
prompt_condition = self.prior(prompt_condition) prompt_condition = self.prior(prompt_condition)
uc = None uc = None
......
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