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

actually fix prior check

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