Pengwei Li
commited on
Commit
·
eee2abc
1
Parent(s):
524783d
Update README.md
Browse files
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 |
|