Merge pull request XingangPan#72 from bulutthecat/main

Added windows support for GUI
PeacefulHorizon · Jun 27, 2023 · d1ed11c · d1ed11c
2 parents ccd84ff + 9059a77
commit d1ed11c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -3,4 +3,7 @@ torchvision
 Ninja
 gradio
 huggingface_hub
-hf_transfer
+hf_transfer
+pyopengl
+imgui
+glfw
diff --git a/scripts/gui.bat b/scripts/gui.bat
@@ -0,0 +1,12 @@
+@echo off
+python visualizer_drag.py ^
+    checkpoints/stylegan2_lions_512_pytorch.pkl ^
+    checkpoints/stylegan2-ffhq-512x512.pkl ^
+    checkpoints/stylegan2-afhqcat-512x512.pkl ^
+    checkpoints/stylegan2-car-config-f.pkl ^
+    checkpoints/stylegan2_dogs_1024_pytorch.pkl ^
+    checkpoints/stylegan2_horses_256_pytorch.pkl ^
+    checkpoints/stylegan2-cat-config-f.pkl ^
+    checkpoints/stylegan2_elephants_512_pytorch.pkl ^
+    checkpoints/stylegan_human_v2_512.pkl ^
+    checkpoints/stylegan2-lhq-256x256.pkl