Spaces:

firatozdemir
/

OAGen_Linear

Running

firatozdemir commited on Oct 3, 2022

Commit

cc9a06d

•

1 Parent(s): 60eed5d

bugfix

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -14,7 +14,7 @@ class SampleFromGAN:
         z = torch.randn(self.z_shp)
         if self.in_gpu:
             z = z.cuda()
-        ims = G(z, c=None)
         ims = ims[:,0,...]
         return ims

         z = torch.randn(self.z_shp)
         if self.in_gpu:
             z = z.cuda()
+        ims = self.G(z, c=None)
         ims = ims[:,0,...]
         return ims