|
|
|
|
|
|
|
|
|
|
|
from PIL import Image, ImageDraw, ImageFont |
|
|
import wandb |
|
|
import os |
|
|
|
|
|
from dalle_mini.backend import ServiceError, get_images_from_backend |
|
|
from dalle_mini.helpers import captioned_strip |
|
|
|
|
|
os.environ["WANDB_SILENT"] = "true" |
|
|
os.environ["WANDB_CONSOLE"] = "off" |
|
|
|
|
|
|
|
|
id = None |
|
|
run = wandb.init(id=id, |
|
|
entity='wandb', |
|
|
project="hf-flax-dalle-mini", |
|
|
job_type="predictions", |
|
|
resume="allow" |
|
|
) |
|
|
|
|
|
def log_to_wandb(prompts): |
|
|
try: |
|
|
backend_url = os.environ["BACKEND_SERVER"] |
|
|
|
|
|
strips = [] |
|
|
for prompt in prompts: |
|
|
print(f"Getting selections for: {prompt}") |
|
|
selected = get_images_from_backend(prompt, backend_url) |
|
|
strip = captioned_strip(selected, prompt) |
|
|
strips.append(wandb.Image(strip)) |
|
|
wandb.log({"images": strips}) |
|
|
except ServiceError as error: |
|
|
print(f"Service unavailable, status: {error.status_code}") |
|
|
except KeyError: |
|
|
print("Error: BACKEND_SERVER unset") |
|
|
|
|
|
prompts = [ |
|
|
"white snow covered mountain under blue sky during daytime", |
|
|
"aerial view of beach during daytime", |
|
|
"aerial view of beach at night", |
|
|
"an armchair in the shape of an avocado", |
|
|
"a logo of an avocado armchair playing music", |
|
|
"young woman riding her bike trough a forest", |
|
|
"rice fields by the mediterranean coast", |
|
|
"white houses on the hill of a greek coastline", |
|
|
"illustration of a shark with a baby shark", |
|
|
"painting of an oniric forest glade surrounded by tall trees", |
|
|
] |
|
|
|
|
|
log_to_wandb(prompts) |
|
|
|
