aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat generate single logo 部署

Browse Source
This commit is contained in:
zhouchengrong
2024-06-03 11:54:52 +08:00
parent 60e0ffca99
commit 8b2adfbd07
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/service/generate_image/service_generate_single_logo.py
View File

@@ -24,6 +24,7 @@ from app.schemas.generate_image import GenerateSingleLogoImageModel
from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_single_logo from app.service.generate_image.utils.upload_sd_image import upload_png_sd, upload_single_logo
logger = logging.getLogger() logger = logging.getLogger()
logging.getLogger("pika").setLevel(logging.WARNING)
class GenerateSingleLogoImage: class GenerateSingleLogoImage:
@@ -89,7 +90,6 @@ class GenerateSingleLogoImage:
# seed # seed
seed = np.array(self.seed, dtype="object").reshape((-1, 1)) seed = np.array(self.seed, dtype="object").reshape((-1, 1))
print('seed: ', self.seed)
input_seed = grpcclient.InferInput("seed", seed.shape, np_to_triton_dtype(seed.dtype)) input_seed = grpcclient.InferInput("seed", seed.shape, np_to_triton_dtype(seed.dtype))
input_seed.set_data_from_numpy(seed) input_seed.set_data_from_numpy(seed)