This commit is contained in:
Shen-Chenhui 2024-04-04 15:26:56 +08:00
parent d08d52ea03
commit bc095aa492

View file

@ -47,7 +47,6 @@ def main():
device = "cuda" if torch.cuda.is_available() else "cpu"
dtype = to_torch_dtype(cfg.dtype)
set_random_seed(seed=cfg.seed)
prompts = cfg.prompt
# ======================================================