Hugging Face's logo

Spaces:
ManuelHuman
/
SynCDtest
Runtime error

App Files Community
Nupur Kumari commited on
Commit
f16d888
·
1 Parent(s): e0f6273

updated sdk

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🖼
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
- sdk_version: 5.4.0
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
4
  colorFrom: purple
5
  colorTo: red
6
  sdk: gradio
7
+ sdk_version: 5.17.1
8
  app_file: app.py
9
  pinned: false
10
  license: mit
app.py CHANGED
@@ -54,7 +54,7 @@ for key,value in finetuned_path.items():
54
  if 'transformer.base_model.model.' in key:
55
  transformer_dict[key.replace('transformer.base_model.model.', '')] = value
56
  pipeline.transformer.load_state_dict(transformer_dict, strict=False)
57
- # pipeline.to('cuda')
58
  pipeline.enable_vae_slicing()
59
  pipeline.enable_vae_tiling()
60
 
 
54
  if 'transformer.base_model.model.' in key:
55
  transformer_dict[key.replace('transformer.base_model.model.', '')] = value
56
  pipeline.transformer.load_state_dict(transformer_dict, strict=False)
57
+ pipeline.to('cuda')
58
  pipeline.enable_vae_slicing()
59
  pipeline.enable_vae_tiling()
60