fix format

This commit is contained in:
strint 2025-10-21 11:59:59 +08:00
parent 2d010f545c
commit fff56de63c

View File

@ -910,4 +910,4 @@ class UNetModel(nn.Module):
if self.predict_codebook_ids:
return self.id_predictor(h)
else:
return self.out(h)
return self.out(h)