fixes imports

stylegan3/legacy.py → legacy.py

@@ -14,8 +14,8 @@ import re
 import copy
 import numpy as np
 import torch
-from stylegan3 import dnnlib
-from stylegan3.torch_utils import misc
+import dnnlib
+from torch_utils import misc
 
 #----------------------------------------------------------------------------
 

stylegan3/__init__.py

@@ -1,2 +0,0 @@
-from stylegan3.dnnlib.util import open_url
-from stylegan3.legacy import load_network_pkl

{stylegan3/torch_utils → torch_utils}/misc.py

@@ -11,7 +11,7 @@ import contextlib
 import numpy as np
 import torch
 import warnings
-from stylegan3 import dnnlib
+import dnnlib
 
 #----------------------------------------------------------------------------
 # Cached construction of constant tensors. Avoids CPU=>GPU copy when the

visualize.py

@@ -5,7 +5,8 @@ import random
 import torch
 from moviepy.audio.AudioClip import AudioArrayClip
 from tqdm import tqdm
-import stylegan3
+import dnnlib
+import legacy
 
 target_sr = 22050
 
@@ -64,8 +65,8 @@ def visualize(audio_file,
 
     # Load pre-trained model
     device = torch.device('cuda')
-    with stylegan3.open_url(network) as f:
-        G = stylegan3.load_network_pkl(f)['G_ema'].to(device) # type: ignore
+    with dnnlib.util.open_url(network) as f:
+        G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
         G.eval()
 
     with torch.no_grad():