🐛 Style Transfer: kill immortal fast gen server at the end of the pro…

…cess
arkhn · Feb 28, 2024 · 38196e3 · 38196e3
1 parent 2b1f1da
commit 38196e3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/style-transfer/style_transfer/gen.py b/lib/style-transfer/style_transfer/gen.py
@@ -127,6 +127,7 @@ def add_prompt(data_point):
 
     wandb.log({"test_dataset": wandb.Table(dataframe=pd.concat(test_dataset))})
     wandb.finish()
+    client.terminate_server()
 
 
 if __name__ == "__main__":