Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -71,6 +71,7 @@ class ARPGModel(DiffusionPipeline):
|
|
| 71 |
seed = kwargs.get("seed", 6)
|
| 72 |
torch.manual_seed(seed)
|
| 73 |
np.random.seed(seed)
|
|
|
|
| 74 |
|
| 75 |
num_steps = kwargs.get("num_steps", 64)
|
| 76 |
cfg_scale = kwargs.get("cfg_scale", 4)
|
|
|
|
| 71 |
seed = kwargs.get("seed", 6)
|
| 72 |
torch.manual_seed(seed)
|
| 73 |
np.random.seed(seed)
|
| 74 |
+
random.seed(seed)
|
| 75 |
|
| 76 |
num_steps = kwargs.get("num_steps", 64)
|
| 77 |
cfg_scale = kwargs.get("cfg_scale", 4)
|