소스 검색

Simplify a function

oobabooga 2 년 전
부모
커밋
0b458bf82d
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      modules/models.py

+ 1 - 2
modules/models.py

@@ -194,8 +194,7 @@ def unload_model():
 
 
 def reload_model():
-    shared.model = shared.tokenizer = None
-    clear_torch_cache()
+    unload_model()
     shared.model, shared.tokenizer = load_model(shared.model_name)