update readme, random seed if none specified

This commit is contained in:
Brett Kuprel
2022-07-01 18:50:11 -04:00
parent ec9e22e4d4
commit fff44d683e
2 changed files with 22 additions and 26 deletions

View File

@@ -6,6 +6,7 @@ from torch import LongTensor
import torch
import json
import requests
import random
torch.set_grad_enabled(False)
torch.set_num_threads(os.cpu_count())
@@ -167,6 +168,7 @@ class MinDalleTorch:
if not self.is_reusable: self.init_decoder()
print("sampling image tokens")
if seed < 0: seed = random.randint(0, 2 ** 31)
torch.manual_seed(seed)
image_tokens = self.decoder.forward(text_tokens, encoder_state)
if not self.is_reusable: del self.decoder