Spaces:

keysun89
/

Diff_Hindi

Runtime error

keysun89 commited on 9 days ago

Commit

e7dee27

verified ·

1 Parent(s): 15c882d

Update unet.py

Files changed (1) hide show

unet.py CHANGED Viewed

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

         if self.predict_codebook_ids:
             return self.id_predictor(h)
         else:
+            return self.out(h)