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 |
|