File size: 892 Bytes
a23872f
 
 
 
 
e0f92a0
 
 
a23872f
e0f92a0
 
 
a23872f
e0f92a0
 
a23872f
e0f92a0
 
 
a23872f
e0f92a0
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import os
import sys

import matplotlib.pyplot as plt
import torch
from backend import ImagePromptEditor, ImageState, ProcessorGradientFlow
from loaders import load_default
from transformers import CLIPModel

if __name__ == "__main__":
    sys.path.append("taming-transformers")
    device = "cuda"

    vqgan = load_default(device)
    vqgan.eval()

    processor = ProcessorGradientFlow(device=device)
    clip = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
    clip.to(device)

    promptoptim = ImagePromptEditor(vqgan, clip, processor, quantize=True)
    state = ImageState(vqgan, promptoptim)
    mask = torch.load("eyebrow_mask.pt")
    x = state.blend("./test_data/face.jpeg", "./test_data/face2.jpeg", 0.5)
    plt.imshow(x)
    plt.show()
    state.apply_prompts(
        "a picture of a woman with big eyebrows", "", 0.009, 40, None, mask=mask
    )
    print("done")