John6666 commited on
Commit
7cd1b41
1 Parent(s): eb9cbd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -309,8 +309,9 @@ if __name__ == "__main__":
309
  }
310
  models = {}
311
  if opt.device == "cuda":
312
- torch.backends.cuda.enable_mem_efficient_sdp(True)
313
- torch.backends.cuda.enable_flash_sdp(True)
 
314
  for name, (repo_id, path, config) in models_info.items():
315
  model_path = hf_hub_download_retry(repo_id=repo_id, filename=f"{path}model.ckpt")
316
  model = MIDIModel(config=MIDIModelConfig.from_name(config))
 
309
  }
310
  models = {}
311
  if opt.device == "cuda":
312
+ pass
313
+ #torch.backends.cuda.enable_mem_efficient_sdp(True)
314
+ #torch.backends.cuda.enable_flash_sdp(True)
315
  for name, (repo_id, path, config) in models_info.items():
316
  model_path = hf_hub_download_retry(repo_id=repo_id, filename=f"{path}model.ckpt")
317
  model = MIDIModel(config=MIDIModelConfig.from_name(config))