Commit 69307367 authored by novelailab's avatar novelailab

fix seed 0 setting bug

parent 6fabb64f
...@@ -71,7 +71,7 @@ class StableDiffusionModel(nn.Module): ...@@ -71,7 +71,7 @@ class StableDiffusionModel(nn.Module):
@torch.no_grad() @torch.no_grad()
def sample(self, request): def sample(self, request):
request = DotMap(request) request = DotMap(request)
if request.seed: if request.seed is not None:
torch.manual_seed(request.seed) torch.manual_seed(request.seed)
np.random.seed(request.seed) np.random.seed(request.seed)
......
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