Pengwei Li commited on
Commit
eee2abc
1 Parent(s): 524783d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -2
README.md CHANGED
@@ -48,8 +48,6 @@ models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
48
  )
49
  #model = models[0].cpu()
50
  #cfg["task"].cpu = True
51
- model = models[0].gpu()
52
- cfg["task"].gpu = True
53
  generator = task.build_generator([model], cfg)
54
 
55
 
 
48
  )
49
  #model = models[0].cpu()
50
  #cfg["task"].cpu = True
 
 
51
  generator = task.build_generator([model], cfg)
52
 
53