Spaces:

nupurkmr9
/

SynCD

Running on Zero

Nupur Kumari commited on 1 day ago

Commit

f16d888

1 Parent(s): e0f6273

updated sdk

Files changed (2) hide show

README.md CHANGED Viewed

@@ -4,7 +4,7 @@ emoji: 🖼
 colorFrom: purple
 colorTo: red
 sdk: gradio
-sdk_version: 5.4.0
 app_file: app.py
 pinned: false
 license: mit

 colorFrom: purple
 colorTo: red
 sdk: gradio
+sdk_version: 5.17.1
 app_file: app.py
 pinned: false
 license: mit

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ for key,value in finetuned_path.items():
     if 'transformer.base_model.model.' in key:
         transformer_dict[key.replace('transformer.base_model.model.', '')] = value
 pipeline.transformer.load_state_dict(transformer_dict, strict=False)
-# pipeline.to('cuda')
 pipeline.enable_vae_slicing()
 pipeline.enable_vae_tiling()

     if 'transformer.base_model.model.' in key:
         transformer_dict[key.replace('transformer.base_model.model.', '')] = value
 pipeline.transformer.load_state_dict(transformer_dict, strict=False)
+pipeline.to('cuda')
 pipeline.enable_vae_slicing()
 pipeline.enable_vae_tiling()