Spaces:
Running
on
Zero
Running
on
Zero
tight-inversion
commited on
Commit
·
fb46df5
1
Parent(s):
896cebe
Make sure seed is in valid range
Browse files- pulid/utils.py +1 -0
pulid/utils.py
CHANGED
|
@@ -14,6 +14,7 @@ from transformers import PretrainedConfig
|
|
| 14 |
|
| 15 |
|
| 16 |
def seed_everything(seed):
|
|
|
|
| 17 |
os.environ["PL_GLOBAL_SEED"] = str(seed)
|
| 18 |
random.seed(seed)
|
| 19 |
np.random.seed(seed)
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
def seed_everything(seed):
|
| 17 |
+
seed = int(seed) % (2**32) # Ensure seed is in the valid range
|
| 18 |
os.environ["PL_GLOBAL_SEED"] = str(seed)
|
| 19 |
random.seed(seed)
|
| 20 |
np.random.seed(seed)
|