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

adjust create_string_buffer that could be measured

Browse Source
This commit is contained in:
henryruhs
2026-05-13 15:31:54 +02:00
parent 832d954df6
commit 912d7eaa52
4 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/audio_encoder.py
+2 -2
View File
@@ -19,9 +19,9 @@ def encode_opus_buffer(opus_encoder : OpusEncoder, input_buffer : bytes, frame_s
output_buffer = b''
if opus_library:
temp_buffer = ctypes.create_string_buffer(4000)
temp_buffer = ctypes.create_string_buffer(2048)
encode_buffer = ctypes.cast(ctypes.create_string_buffer(input_buffer), ctypes.POINTER(ctypes.c_float))
encode_length = opus_library.opus_encode_float(opus_encoder, encode_buffer, frame_size, temp_buffer, 4000)
encode_length = opus_library.opus_encode_float(opus_encoder, encode_buffer, frame_size, temp_buffer, 2048)
if encode_length:
output_buffer = temp_buffer.raw[:encode_length]
facefusion/rtc.py
+2 -2
View File
@@ -112,7 +112,7 @@ def add_video_track(peer_connection : PeerConnection, media_direction : MediaDir
def create_sdp(peer_connection : PeerConnection) -> Optional[SdpOffer]:
datachannel_library = datachannel_module.create_static_library()
datachannel_library.rtcSetLocalDescription(peer_connection, b'offer')
buffer_size = 16384
buffer_size = 8192
buffer_string = ctypes.create_string_buffer(buffer_size)
if datachannel_library.rtcGetLocalDescription(peer_connection, buffer_string, buffer_size) > 0:
@@ -145,7 +145,7 @@ def negotiate_sdp(peer_connection : PeerConnection, sdp_offer : SdpOffer) -> Opt
datachannel_library.rtcSetRemoteDescription(peer_connection, sdp_offer.encode(), b'offer')
sdp_event.wait(timeout = 5)
sdp_buffer_size = 16384
sdp_buffer_size = 8192
sdp_buffer = ctypes.create_string_buffer(sdp_buffer_size)
if datachannel_library.rtcGetLocalDescription(peer_connection, sdp_buffer, sdp_buffer_size) > 0:
facefusion/system.py
+4 -4
View File
@@ -38,15 +38,15 @@ def detect_nvidia_graphic_devices() -> List[GraphicDevice]:
if nvidia_ml_library:
nvidia_ml_library.nvmlInit_v2()
driver_version = ctypes.create_string_buffer(80)
nvidia_ml_library.nvmlSystemGetDriverVersion(driver_version, 80)
driver_version = ctypes.create_string_buffer(128)
nvidia_ml_library.nvmlSystemGetDriverVersion(driver_version, 128)
cuda_version = ctypes.c_int()
nvidia_ml_library.nvmlSystemGetCudaDriverVersion(ctypes.byref(cuda_version))
for device_handle in nvidia_ml_module.find_device_handles(nvidia_ml_library):
device_name = ctypes.create_string_buffer(96)
nvidia_ml_library.nvmlDeviceGetName(device_handle, device_name, 96)
device_name = ctypes.create_string_buffer(128)
nvidia_ml_library.nvmlDeviceGetName(device_handle, device_name, 128)
device_memory = nvidia_ml_module.define_device_memory()
nvidia_ml_library.nvmlDeviceGetMemoryInfo(device_handle, ctypes.byref(device_memory))
facefusion/video_encoder.py
+3 -3
View File
@@ -10,10 +10,10 @@ def create_vpx_encoder(frame_resolution : Resolution, bitrate : BitRate, thread_
vpx_library = vpx_module.create_static_library()
if vpx_library:
vpx_encoder = ctypes.create_string_buffer(512)
vpx_encoder = ctypes.create_string_buffer(64)
vp8_codec = ctypes.c_void_p.in_dll(vpx_library, 'vpx_codec_vp8_cx_algo')
config_buffer = ctypes.create_string_buffer(4096)
config_buffer = ctypes.create_string_buffer(512)
if vpx_library.vpx_codec_enc_config_default(ctypes.byref(vp8_codec), config_buffer, 0) == 0:
struct.pack_into('I', config_buffer, 4, thread_count)
@@ -42,7 +42,7 @@ def encode_vpx_buffer(vpx_encoder : VpxEncoder, input_buffer : bytes, frame_reso
output_buffer = b''
if vpx_library:
temp_buffer = ctypes.create_string_buffer(512)
temp_buffer = ctypes.create_string_buffer(256)
encode_buffer = ctypes.create_string_buffer(input_buffer)
if vpx_library.vpx_img_wrap(temp_buffer, 0x102, frame_resolution[0], frame_resolution[1], 1, encode_buffer) and vpx_library.vpx_codec_encode(vpx_encoder, temp_buffer, frame_index, 1, 0, 1) == 0: