Commit 69307367 authored by novelailab's avatar novelailab

fix seed 0 setting bug

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