joacorf33 commited on
Commit
d3e3435
1 Parent(s): 5a96139

huggan issue fixed

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -5,7 +5,7 @@ import numpy as np
5
  # Pythorch
6
  import torch
7
  # A hugging face library
8
- from huggan.pytorch.lightweight_gan.leightweight_gan import LightweightGAN
9
 
10
  """Let's now set up the model functions"""
11
 
@@ -25,3 +25,4 @@ def generate(gan, batch_size=1):
25
  #
26
  ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
27
  return ims
 
 
5
  # Pythorch
6
  import torch
7
  # A hugging face library
8
+ from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
9
 
10
  """Let's now set up the model functions"""
11
 
 
25
  #
26
  ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
27
  return ims
28
+