fix a bug caused by merge

This commit is contained in:
AUTOMATIC 2023-01-13 15:04:37 +03:00
parent 1849f6eb80
commit 82725f0ac4

View File

@ -11,6 +11,7 @@ import datetime
import csv import csv
import safetensors.torch import safetensors.torch
import numpy as np
from PIL import Image, PngImagePlugin from PIL import Image, PngImagePlugin
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter