add DiffusionInpaintModel

2023-01-27 20:59:22 +08:00
parent 96659f2aef
commit 205170e1e5
5 changed files with 55 additions and 122 deletions
--- a/lama_cleaner/model/utils.py
+++ b/lama_cleaner/model/utils.py
@@ -1,4 +1,5 @@
 import math
+import random
 from typing import Any

 import torch
@@ -713,3 +714,10 @@ def torch_gc():
    if torch.cuda.is_available():
        torch.cuda.empty_cache()
        torch.cuda.ipc_collect()
+
+
+def set_seed(seed: int):
+    random.seed(seed)
+    np.random.seed(seed)
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed_all(seed)