Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
abreza
/
SadTalker
like
23
Running
on
Zero
App
Files
Files
Community
4
f599b88
SadTalker
/
src
/
utils
/
safetensor_helper.py
abreza
init
8f6558d
5 months ago
raw
Copy download link
history
blame
Safe
198 Bytes
def
load_x_from_safetensor
(
checkpoint, key
):
x_generator = {}
for
k,v
in
checkpoint.items():
if
key
in
k:
x_generator[k.replace(key+
'.'
,
''
)] = v
return
x_generator