Update README.md

README.md

@@ -279,8 +279,9 @@ import torch.nn as nn
 from models import build_model
 from PIL import Image
 
-random.seed(42)
-torch.manual_seed(42)
+seed = 1234
+random.seed(seed)
+torch.manual_seed(seed)
 
 # The local directory you save the LaVIT pre-trained weight, 
 # it will automatically download the checkpoint from huggingface
@@ -320,10 +321,15 @@ For the Image generation, the Classifier-Free Guidance scale is important. A lar
 ```python
 import os
 import torch
+import random
 import torch.nn as nn
 from models import build_model
 from PIL import Image
 
+seed = 1234
+random.seed(seed)
+torch.manual_seed(seed)
+
 # The local directory you save the LaVIT pre-trained weight, 
 # it will automatically download the checkpoint from huggingface
 model_path = '/path/LaVIT_weight'