Spaces:
Sleeping
Sleeping
Singularity666
commited on
Update main.py
Browse files
main.py
CHANGED
@@ -10,6 +10,7 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler, UNet2DConditionMod
|
|
10 |
from transformers import CLIPTextModel, CLIPTokenizer
|
11 |
from accelerate import Accelerator
|
12 |
from tqdm.auto import tqdm
|
|
|
13 |
|
14 |
class CustomDataset(Dataset):
|
15 |
def __init__(self, data_dir, prompt, tokenizer, size=512, center_crop=False):
|
|
|
10 |
from transformers import CLIPTextModel, CLIPTokenizer
|
11 |
from accelerate import Accelerator
|
12 |
from tqdm.auto import tqdm
|
13 |
+
from PIL import Image
|
14 |
|
15 |
class CustomDataset(Dataset):
|
16 |
def __init__(self, data_dir, prompt, tokenizer, size=512, center_crop=False):
|