From 9adf754a6b3e78e3e1dbae83fa795c691f474e75 Mon Sep 17 00:00:00 2001 From: taabata <57796911+taabata@users.noreply.github.com> Date: Sat, 28 Dec 2024 04:02:31 +0300 Subject: [PATCH] Delete SANA/getembeds.py --- SANA/getembeds.py | 39 --------------------------------------- 1 file changed, 39 deletions(-) delete mode 100644 SANA/getembeds.py diff --git a/SANA/getembeds.py b/SANA/getembeds.py deleted file mode 100644 index 9862586..0000000 --- a/SANA/getembeds.py +++ /dev/null @@ -1,39 +0,0 @@ -import requests, json, argparse -import numpy as np -import torch -from pipes.sana_img2img import SanaPipelineImg2Img - - -parser = argparse.ArgumentParser() - -parser.add_argument("--prompt","-p") -parser.add_argument("--model","-m") - -args = parser.parse_args() - -model_name = args.model - -pipe = SanaPipelineImg2Img.from_pretrained( - model_name, - torch_dtype=torch.bfloat16, - transformer=None, - vae=None - -) -pipe.to('cpu') -prompt = args.prompt -embeds = pipe( - prompt=prompt, - height=512, - width=512, - guidance_scale=5.0, - num_inference_steps=2, - textonly=True -) - - -data = {} -for i in embeds: - embs = json.dumps(embeds[i].type(torch.float32).numpy().astype(np.float32).tolist()) - data[i] = embs -req = requests.post("http://127.0.0.1:5000/getEmbeds",json=data,headers={'accept': 'content_type_value'})