more
This commit is contained in:
parent
c6f4c1621b
commit
c62e263349
2 changed files with 9 additions and 7 deletions
13
generate.py
13
generate.py
|
|
@ -686,7 +686,7 @@ def wan_reference_to_video(foreground, background, keep=False):
|
||||||
background_url,
|
background_url,
|
||||||
],
|
],
|
||||||
"aspect_ratio": "16:9",
|
"aspect_ratio": "16:9",
|
||||||
"resolution": "1080p",
|
"resolution": "720p",
|
||||||
"enable_prompt_expansion": False,
|
"enable_prompt_expansion": False,
|
||||||
"multi_shots": True,
|
"multi_shots": True,
|
||||||
"enable_safety_checker": False
|
"enable_safety_checker": False
|
||||||
|
|
@ -788,7 +788,7 @@ def ltx_a2v(item, character, prompt=None, first_frame=None, keep=False):
|
||||||
ox.net.save_url(output_url, output, overwrite=True)
|
ox.net.save_url(output_url, output, overwrite=True)
|
||||||
ai = add_ai_variant(item, output, "ai:audio-to-video")
|
ai = add_ai_variant(item, output, "ai:audio-to-video")
|
||||||
ai.data["model"] = model
|
ai.data["model"] = model
|
||||||
ai.data["prompt"] = prompt
|
ai.data["prompt"] = ox.escape_html(prompt)
|
||||||
ai.data["seed"] = result["seed"]
|
ai.data["seed"] = result["seed"]
|
||||||
ai.save()
|
ai.save()
|
||||||
first_frame.add(ai)
|
first_frame.add(ai)
|
||||||
|
|
@ -930,12 +930,12 @@ def describe_item(item, neutral=False):
|
||||||
if isinstance(item, str):
|
if isinstance(item, str):
|
||||||
item = Item.objects.get(public_id=item)
|
item = Item.objects.get(public_id=item)
|
||||||
if item.get("prompt") and neutral:
|
if item.get("prompt") and neutral:
|
||||||
return item.get("prompt")
|
return ox.decode_html(item.get("prompt"))
|
||||||
video_url = public_video_url(item)
|
video_url = public_video_url(item)
|
||||||
prompt = describe_video(video_url, neutral)
|
prompt = describe_video(video_url, neutral)
|
||||||
if neutral:
|
if neutral:
|
||||||
item.refresh_from_db()
|
item.refresh_from_db()
|
||||||
item.data["prompt"] = prompt
|
item.data["prompt"] = ox.escape_html(prompt)
|
||||||
item.save()
|
item.save()
|
||||||
return prompt
|
return prompt
|
||||||
|
|
||||||
|
|
@ -1510,6 +1510,9 @@ def process_reshoot_firstframe(character='P1'):
|
||||||
item.refresh_from_db()
|
item.refresh_from_db()
|
||||||
add_tag(item, 'ai-failed')
|
add_tag(item, 'ai-failed')
|
||||||
print('>> failed', item)
|
print('>> failed', item)
|
||||||
|
continue
|
||||||
|
if not first_frame:
|
||||||
|
continue
|
||||||
if first_frame.items.filter(data__type__icontains='ai:').exists():
|
if first_frame.items.filter(data__type__icontains='ai:').exists():
|
||||||
continue
|
continue
|
||||||
print(item, first_frame)
|
print(item, first_frame)
|
||||||
|
|
@ -1527,7 +1530,7 @@ def process_motion_firstframe(character="P1", keep=False):
|
||||||
if ai.exists() or 'ai-failed' in item.data.get('tags', []):
|
if ai.exists() or 'ai-failed' in item.data.get('tags', []):
|
||||||
print('>> skip', item)
|
print('>> skip', item)
|
||||||
continue
|
continue
|
||||||
print(i)
|
print(item)
|
||||||
try:
|
try:
|
||||||
replace_character_motion_control(item, character, keep=keep)
|
replace_character_motion_control(item, character, keep=keep)
|
||||||
except:
|
except:
|
||||||
|
|
|
||||||
|
|
@ -98,7 +98,7 @@ def compose(clips, fragment, target=150, base=1024, voice_over=None, options=Non
|
||||||
ai_clips.append(clip)
|
ai_clips.append(clip)
|
||||||
else:
|
else:
|
||||||
non_ai_clips.append(clip)
|
non_ai_clips.append(clip)
|
||||||
if ai_length < target * 0.6 and ai_clips:
|
if ai_length < target * 0.7 and ai_clips:
|
||||||
clip = random_choice(seq, ai_clips, pop=True)
|
clip = random_choice(seq, ai_clips, pop=True)
|
||||||
clip["use_ai"] = True
|
clip["use_ai"] = True
|
||||||
selected_clips.append(clip)
|
selected_clips.append(clip)
|
||||||
|
|
@ -1059,7 +1059,6 @@ def generate_clips(options):
|
||||||
if '-a-t' in variant:
|
if '-a-t' in variant:
|
||||||
b_variant = variant.replace('-a-t', '-b-t').split('-t')[0]
|
b_variant = variant.replace('-a-t', '-b-t').split('-t')[0]
|
||||||
for old in voice_over[fragment][type]:
|
for old in voice_over[fragment][type]:
|
||||||
print(type(old))
|
|
||||||
if isinstance(old, list) and old[0]['variant'].startswith(b_variant):
|
if isinstance(old, list) and old[0]['variant'].startswith(b_variant):
|
||||||
old.insert(0, vo_variant)
|
old.insert(0, vo_variant)
|
||||||
done = True
|
done = True
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue