jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify set_audio_sample_size()

Browse Source
This commit is contained in:
henryruhs
2025-01-26 04:16:28 +01:00
parent b9eb6a7cc0
commit 02d11435a6
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
facefusion/ffmpeg_builder.py
View File

@@ -114,9 +114,9 @@ def set_audio_sample_rate(audio_sample_rate : int) -> Commands:
def set_audio_sample_size(audio_sample_size : int) -> Commands: def set_audio_sample_size(audio_sample_size : int) -> Commands:
if audio_sample_size == 16: if audio_sample_size == 16:
return [ '-f', 's16le', '-acodec', 'pcm_s16le' ] return [ '-f', 's16le' ]
if audio_sample_size == 32: if audio_sample_size == 32:
return [ '-f', 's32le', '-acodec', 'pcm_s32le' ] return [ '-f', 's32le' ]
return [] return []

8
tests/test_ffmpeg_builder.py
View File

@@ -12,7 +12,7 @@ def test_chain() -> None:
assert chain(ffmpeg_builder.set_progress()) == [ '-progress' ] assert chain(ffmpeg_builder.set_progress()) == [ '-progress' ]
def test_stream_mode() -> None: def test_set_stream_mode() -> None:
assert set_stream_mode('udp') == [ '-f', 'mpegts' ] assert set_stream_mode('udp') == [ '-f', 'mpegts' ]
assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ] assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ]
@@ -24,9 +24,9 @@ def test_select_frame_range() -> None:
assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ] assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ]
def test_audio_sample_size() -> None: def test_set_audio_sample_size() -> None:
assert set_audio_sample_size(16) == [ '-f', 's16le', '-acodec', 'pcm_s16le' ] assert set_audio_sample_size(16) == [ '-f', 's16le' ]
assert set_audio_sample_size(32) == [ '-f', 's32le', '-acodec', 'pcm_s32le' ] assert set_audio_sample_size(32) == [ '-f', 's32le' ]
def test_set_audio_quality() -> None: def test_set_audio_quality() -> None: