Hugging Face's logo

Spaces:
mehdidc
/
text_to_image_ddgan
Runtime error

App Files Community
Mehdi Cherti commited on
Commit
e30ec05
·
1 Parent(s): 1e5aadc

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def download(filename):
16
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
17
  models = {
18
  "diffusion_db_128ch_1timesteps_openclip_vith14": load_model(get_model_config('ddgan_ddb_v2'), download('diffusion_db_128ch_1timesteps_openclip_vith14.th'), device=device),
19
- "diffusion_db_192ch_2timesteps_openclip_vith14": load_model(get_model_config('ddgan_ddb_v3'), download('diffusion_db_192ch_2timesteps_openclip_vith14.th'), device=device),
20
  }
21
  default = "diffusion_db_128ch_1timesteps_openclip_vith14"
22
 
 
16
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
17
  models = {
18
  "diffusion_db_128ch_1timesteps_openclip_vith14": load_model(get_model_config('ddgan_ddb_v2'), download('diffusion_db_128ch_1timesteps_openclip_vith14.th'), device=device),
19
+ #"diffusion_db_192ch_2timesteps_openclip_vith14": load_model(get_model_config('ddgan_ddb_v3'), download('diffusion_db_192ch_2timesteps_openclip_vith14.th'), device=device),
20
  }
21
  default = "diffusion_db_128ch_1timesteps_openclip_vith14"
22