Fix clean_models (#356)

2023-06-14 14:22:58 +02:00 · 2023-06-14 14:22:58 +02:00 · 9e41b5bd44
parent f6f2db527b
commit 9e41b5bd44
1 changed files with 1 additions and 1 deletions
--- a/bark/generation.py
+++ b/bark/generation.py
@ -184,7 +184,7 @@ def _clear_cuda_cache():

 def clean_models(model_key=None):
    global models
-    model_keys = [model_key] if model_key is not None else models.keys()
+    model_keys = [model_key] if model_key is not None else list(models.keys())
    for k in model_keys:
        if k in models:
            del models[k]