Spaces:
Runtime error
Runtime error
cosmicdream
commited on
Commit
•
2fda7e8
1
Parent(s):
9775aa2
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
30 |
#device = "cuda"
|
31 |
print('Using device:', device)
|
32 |
|
33 |
-
|
34 |
tokenizer = get_tokenizer()
|
35 |
vae = get_vae(dwt=True).to(device)
|
36 |
|
@@ -54,8 +54,8 @@ text = 'радуга на фоне ночного города'
|
|
54 |
def model(text, tokenizer=tokenizer, dalle=dalle, vae=vae, top_k=2048, images_num=1, bs=8, top_p=0.995):
|
55 |
# _pil_images, _scores = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
56 |
pil_images = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
57 |
-
pil_images = pil_images[0]
|
58 |
-
return pil_images
|
59 |
|
60 |
seed_everything(42)
|
61 |
pil_images = []
|
|
|
30 |
#device = "cuda"
|
31 |
print('Using device:', device)
|
32 |
|
33 |
+
dalle = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
|
34 |
tokenizer = get_tokenizer()
|
35 |
vae = get_vae(dwt=True).to(device)
|
36 |
|
|
|
54 |
def model(text, tokenizer=tokenizer, dalle=dalle, vae=vae, top_k=2048, images_num=1, bs=8, top_p=0.995):
|
55 |
# _pil_images, _scores = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
56 |
pil_images = generate_images(text, tokenizer, dalle, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
|
57 |
+
# pil_images = pil_images[0]
|
58 |
+
return pil_images[0]
|
59 |
|
60 |
seed_everything(42)
|
61 |
pil_images = []
|