允许 model.generate 使用bytes io, 以便不写入文件，节省io时间 (#2343)

maliubiao

2024-12-29 172a3152b42af36443ec6a0a39969471c35b893d

1
2
3
4
5
6
7
8
9
10
import random
 
import numpy as np
import torch
 
 
def set_all_random_seed(seed: int):
    random.seed(seed)
    np.random.seed(seed)
    torch.random.manual_seed(seed)