Paillat-dev/viralfactory
1
0
Fork 0
mirror of https://github.com/Paillat-dev/viralfactory.git synced 2026-01-02 01:06:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Paillat
2024-02-22 15:14:36 +01:00
parent abef1dcb6b
commit d08a2c29ef
2 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/engines/AssetsEngine/AssetsEngineSelector.py
View File

@@ -30,7 +30,8 @@ class AssetsEngineSelector:
)["assets"] )["assets"]
clips: list = [] clips: list = []
for engine in self.ctx.assetsengine: for engine in self.ctx.assetsengine:
assets_opts = [asset for asset in assets if asset["engine"] == engine.name] assets_opts = [
assets_opts = [asset["args"] for asset in assets_opts] asset["args"] for asset in assets if asset["engine"] == engine.name
]
clips.extend(engine.get_assets(assets_opts)) clips.extend(engine.get_assets(assets_opts))
self.ctx.index_3.extend(clips) self.ctx.index_3.extend(clips)

18
src/engines/ScriptEngine/ShowerThoughtsScriptEngine.py
View File

@@ -23,17 +23,13 @@ class ShowerThoughtsScriptEngine(BaseScriptEngine):
) )
sys_prompt = sys_prompt.format(n_sentences=self.n_sentences) sys_prompt = sys_prompt.format(n_sentences=self.n_sentences)
chat_prompt = chat_prompt.format(n_sentences=self.n_sentences) chat_prompt = chat_prompt.format(n_sentences=self.n_sentences)
self.ctx.script = ( self.ctx.script = self.ctx.powerfulllmengine.generate(
self.ctx.powerfulllmengine.generate( system_prompt=sys_prompt,
system_prompt=sys_prompt, chat_prompt=chat_prompt,
chat_prompt=chat_prompt, max_tokens=20 * self.n_sentences,
max_tokens=20 * self.n_sentences, temperature=1.3,
temperature=1.3, json_mode=False,
json_mode=False, ).strip()
)
.strip()
.copy()
)
@classmethod @classmethod
def get_options(cls) -> list: def get_options(cls) -> list: