Spaces:
Runtime error
Runtime error
detach cambio
Browse files
utils.py
CHANGED
|
@@ -10,6 +10,6 @@ def carga_modelo(modelo_name ="ceyda/butterfly_cropped_uniq1K_512", model_versio
|
|
| 10 |
def genera(gan, batch_size =1):
|
| 11 |
with torch.no_grad():
|
| 12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0)*255
|
| 13 |
-
ims = ims.permute(0,2,3,1).
|
| 14 |
|
| 15 |
return ims
|
|
|
|
| 10 |
def genera(gan, batch_size =1):
|
| 11 |
with torch.no_grad():
|
| 12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0)*255
|
| 13 |
+
ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8)
|
| 14 |
|
| 15 |
return ims
|