hysts HF staff commited on
Commit
3a532cb
1 Parent(s): 90f6702
Files changed (1) hide show
  1. model.py +0 -1
model.py CHANGED
@@ -15,7 +15,6 @@ HF_TOKEN = os.environ['HF_TOKEN']
15
 
16
 
17
  class Model:
18
-
19
  def __init__(self, device: str | torch.device):
20
  self.device = torch.device(device)
21
  self.model = self.load_model('stylegan_human_v2_1024.pkl')
 
15
 
16
 
17
  class Model:
 
18
  def __init__(self, device: str | torch.device):
19
  self.device = torch.device(device)
20
  self.model = self.load_model('stylegan_human_v2_1024.pkl')