Update src/load_models.py
Browse files- src/load_models.py +1 -1
src/load_models.py
CHANGED
|
@@ -8,7 +8,7 @@ from huggingface_hub import hf_hub_download
|
|
| 8 |
# Point to your local stylegan2 utilities
|
| 9 |
#sys.path.append("src/stylegan2") # Adjust if the path differs
|
| 10 |
|
| 11 |
-
|
| 12 |
|
| 13 |
# Globals
|
| 14 |
stylegan_generator = None
|
|
|
|
| 8 |
# Point to your local stylegan2 utilities
|
| 9 |
#sys.path.append("src/stylegan2") # Adjust if the path differs
|
| 10 |
|
| 11 |
+
import legacy
|
| 12 |
|
| 13 |
# Globals
|
| 14 |
stylegan_generator = None
|