startup message formatting

This commit is contained in:
Juha Jeronen
2024-01-10 17:26:24 +02:00
parent 8f71af9e48
commit f0a311cbfe

View File

@@ -965,7 +965,7 @@ if "talkinghead" in modules:
if model == "auto": # default
# FP16 boosts the rendering performance by ~1.5x, but is only supported on GPU.
model = "separable_half" if args.talkinghead_gpu else "separable_float"
print(f"Initializing talkinghead pipeline in {Fore.GREEN}{Style.BRIGHT}{mode}{Style.RESET_ALL} mode with model {Fore.GREEN}{Style.BRIGHT}{model}{Style.RESET_ALL}...")
print(f"Initializing {Fore.GREEN}{Style.BRIGHT}talkinghead{Style.RESET_ALL} pipeline in {Fore.GREEN}{Style.BRIGHT}{mode}{Style.RESET_ALL} mode with model {Fore.GREEN}{Style.BRIGHT}{model}{Style.RESET_ALL}...")
try:
from talkinghead.tha3.app.util import maybe_install_models as talkinghead_maybe_install_models