danhergir commited on
Commit
3d7ea81
1 Parent(s): c51dff8
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -2,6 +2,7 @@ import numpy as np
2
  import torch
3
  from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
 
5
  def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
6
  gan = LightweightGAN.from_pretrained(model_name, version=model_version)
7
  gan.eval()
 
2
  import torch
3
  from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
5
+
6
  def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
7
  gan = LightweightGAN.from_pretrained(model_name, version=model_version)
8
  gan.eval()