diff --git a/tti-db/db-data/init.sql b/tti-db/db-data/init.sql index bab46f5..23964d3 100644 --- a/tti-db/db-data/init.sql +++ b/tti-db/db-data/init.sql @@ -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: 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'); + + diff --git a/tti-provider/requirements.txt b/tti-provider/requirements.txt index bbdccde..823d5bd 100644 --- a/tti-provider/requirements.txt +++ b/tti-provider/requirements.txt @@ -4,6 +4,6 @@ uvicorn validators diffusers -# transformers -# accelerate +transformers +accelerate torch diff --git a/tti-provider/runner.py b/tti-provider/runner.py index c08b052..0c953e0 100644 --- a/tti-provider/runner.py +++ b/tti-provider/runner.py @@ -129,7 +129,7 @@ try: """ img_uuid = str(uuid.uuid4()) - file_name = 'images/' + img_uuid + '.png' + file_name = '/app/images/' + img_uuid + '.png' # logging.debug(file_name) image.save(file_name) @@ -150,7 +150,7 @@ try: val = (row['task_id'],) cursor.execute(sql, val) mydb.commit() - + exit(0) # To prevent more Memory allocation... elif row['callback_send'] == 0: logging.debug("A Task is already running...") exit(0)