fix timestamp issue

This commit is contained in:
Benjamin Berman 2023-08-12 14:29:50 -07:00
parent 130378c449
commit 894aee44bd

View File

@ -46,7 +46,7 @@ def get_good_outputs(t: QueueTuple):
class HistoryEntry(typing.TypedDict):
prompt: QueueTuple
outputs: dict
timestamp: datetime.datetime
timestamp: float
@dataclass
@ -780,7 +780,7 @@ class PromptQueue:
with self.mutex:
queue_item = self.currently_running.pop(item_id)
prompt = queue_item.queue_tuple
self.history[prompt[1]] = {"prompt": prompt, "outputs": {}, "timestamp": datetime.datetime.now()}
self.history[prompt[1]] = {"prompt": prompt, "outputs": {}, "timestamp": datetime.datetime.timestamp()}
for o in outputs:
self.history[prompt[1]]["outputs"][o] = outputs[o]
self.server.queue_updated()