Update app.py

app.py

@@ -27,11 +27,11 @@ import ruclip
 # prepare models:
 #device = 'cuda'
 device = "cuda:0" if torch.cuda.is_available() else "CPU"
-fp16=True
-if device == 'CPU':
-    fp16 = False
+#fp16=True
+#if device == 'CPU':
+#    fp16 = False
 
-model = get_rudalle_model('Malevich', pretrained=True, fp16=fp16, device=device)
+model = get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=True).to(device)
 
@@ -46,13 +46,13 @@ seed_everything(42)
 pil_images = []
 scores = []
 for top_k, top_p, images_num in [
-    (2048, 0.995, 24),
+    (2048, 0.995, 6),
 ]:
     _pil_images, _scores = generate_images(text, tokenizer, model, vae, top_k=top_k, images_num=images_num, bs=8, top_p=top_p)
     pil_images += _pil_images
     scores += _scores
 
-show(pil_images, 6)
+#show(pil_images, 6)
 #TEST--------