Hugging Face's logo

Spaces:
Kunbyte
/
OmniTry
Running on Zero

App Files Community
1
fengyutong commited on
Commit
b94267e
·
verified ·
1 Parent(s): da2f8e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ lora_config = LoraConfig(
49
  transformer.add_adapter(lora_config, adapter_name='vtryon_lora')
50
  transformer.add_adapter(lora_config, adapter_name='garment_lora')
51
 
52
- with safe_open('OmniTry/omnitry_v1_unified_stage2.safetensors', framework="pt") as f:
53
  lora_weights = {k: f.get_tensor(k) for k in f.keys()}
54
  transformer.load_state_dict(lora_weights, strict=False)
55
 
 
49
  transformer.add_adapter(lora_config, adapter_name='vtryon_lora')
50
  transformer.add_adapter(lora_config, adapter_name='garment_lora')
51
 
52
+ with safe_open('OmniTry/omnitry_v1_unified.safetensors', framework="pt") as f:
53
  lora_weights = {k: f.get_tensor(k) for k in f.keys()}
54
  transformer.load_state_dict(lora_weights, strict=False)
55