update
Browse files- infer_api.py +1 -1
infer_api.py
CHANGED
|
@@ -129,7 +129,7 @@ for file in all_files:
|
|
| 129 |
hf_hub_download(repo_id, file, local_dir="./ckpt")
|
| 130 |
|
| 131 |
@spaces.GPU
|
| 132 |
-
def
|
| 133 |
random.seed(seed)
|
| 134 |
np.random.seed(seed)
|
| 135 |
torch.manual_seed(seed)
|
|
|
|
| 129 |
hf_hub_download(repo_id, file, local_dir="./ckpt")
|
| 130 |
|
| 131 |
@spaces.GPU
|
| 132 |
+
def set_seed2(seed):
|
| 133 |
random.seed(seed)
|
| 134 |
np.random.seed(seed)
|
| 135 |
torch.manual_seed(seed)
|