tidoni/tti-provider
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

V0.9.0.1

Browse source
This commit is contained in:
Niklas Müller 2024-02-27 11:45:57 +01:00
parent 7f6da3926e
commit b77113c9d2
3 changed files with 19 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tti-db/db-data/init.sql
View file

@ -66,4 +66,19 @@ INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_st
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: Fooocus Sharp', 'portrait of a beautiful woman looking at the viewer', 'Fooocus Sharp'); INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: Fooocus Sharp', 'portrait of a beautiful woman looking at the viewer', 'Fooocus Sharp');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-bubble bobble', 'portrait of a beautiful woman looking at the viewer', 'game-bubble bobble');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-cyberpunk game', 'portrait of a beautiful woman looking at the viewer', 'game-cyberpunk game');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-fighting game', 'portrait of a beautiful woman looking at the viewer', 'game-fighting game');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-gta', 'portrait of a beautiful woman looking at the viewer', 'game-gta');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-mario', 'portrait of a beautiful woman looking at the viewer', 'game-mario');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-minecraft', 'portrait of a beautiful woman looking at the viewer', 'game-minecraft');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-pokemon', 'portrait of a beautiful woman looking at the viewer', 'game-pokemon');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-retro arcade', 'portrait of a beautiful woman looking at the viewer', 'game-retro arcade');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-retro game', 'portrait of a beautiful woman looking at the viewer', 'game-retro game');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-rpg fantasy game', 'portrait of a beautiful woman looking at the viewer', 'game-rpg fantasy game');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-strategy game', 'portrait of a beautiful woman looking at the viewer', 'game-strategy game');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-streetfighter', 'portrait of a beautiful woman looking at the viewer', 'game-streetfighter');
INSERT INTO `tti_tasks` (model, itterations, file_name, initial_prompt, image_style) VALUES ('stabilityai/sdxl-turbo', 1, 'Art Demo: game-zelda', 'portrait of a beautiful woman looking at the viewer', 'game-zelda');

4
tti-provider/requirements.txt
View file

@ -4,6 +4,6 @@ uvicorn
validators validators
diffusers diffusers
# transformers transformers
# accelerate accelerate
torch torch

4
tti-provider/runner.py
View file

@ -129,7 +129,7 @@ try:
""" """
img_uuid = str(uuid.uuid4()) img_uuid = str(uuid.uuid4())
file_name = 'images/' + img_uuid + '.png' file_name = '/app/images/' + img_uuid + '.png'
# logging.debug(file_name) # logging.debug(file_name)
image.save(file_name) image.save(file_name)
@ -150,7 +150,7 @@ try:
val = (row['task_id'],) val = (row['task_id'],)
cursor.execute(sql, val) cursor.execute(sql, val)
mydb.commit() mydb.commit()
exit(0) # To prevent more Memory allocation...
elif row['callback_send'] == 0: elif row['callback_send'] == 0:
logging.debug("A Task is already running...") logging.debug("A Task is already running...")
exit(0) exit(0)