Hugging Face's logo

Spaces:
SebastianBodza
/
Kartoffelbox_v0.2_preview
Running on Zero

App Files Community
SebastianBodza commited on
Commit
2f43ddf
·
verified ·
1 Parent(s): 09a240e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def get_or_load_model():
35
 
36
  if "cond_enc.spkr_enc.0.weight" in cleaned_state_dict:
37
  cleaned_state_dict["cond_enc.spkr_enc.weight"] = cleaned_state_dict.pop("cond_enc.spkr_enc.0.weight")
38
- if "cond_enc.spkr_enc.0.bias" in t3_state:
39
  cleaned_state_dict["cond_enc.spkr_enc.bias"] = cleaned_state_dict.pop("cond_enc.spkr_enc.0.bias")
40
 
41
  MODEL.t3.load_state_dict(cleaned_state_dict)
 
35
 
36
  if "cond_enc.spkr_enc.0.weight" in cleaned_state_dict:
37
  cleaned_state_dict["cond_enc.spkr_enc.weight"] = cleaned_state_dict.pop("cond_enc.spkr_enc.0.weight")
38
+ if "cond_enc.spkr_enc.0.bias" in cleaned_state_dict:
39
  cleaned_state_dict["cond_enc.spkr_enc.bias"] = cleaned_state_dict.pop("cond_enc.spkr_enc.0.bias")
40
 
41
  MODEL.t3.load_state_dict(cleaned_state_dict)