rriverar75 commited on
Commit
ba00f16
·
1 Parent(s): b14c380

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -1,7 +1,7 @@
1
  import numpy as np
2
  import torch
3
- from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
-
5
 
6
  ## Cargamos el modelo desde el Hub de Hugging Face
7
  def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
 
1
  import numpy as np
2
  import torch
3
+ ##from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
+ from lightweight_gan import LightweightGAN
5
 
6
  ## Cargamos el modelo desde el Hub de Hugging Face
7
  def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):