Update README.md
Browse files
README.md
CHANGED
|
@@ -48,6 +48,8 @@ models, cfg, task = load_model_ensemble_and_task_from_hf_hub(
|
|
| 48 |
)
|
| 49 |
#model = models[0].cpu()
|
| 50 |
#cfg["task"].cpu = True
|
|
|
|
|
|
|
| 51 |
generator = task.build_generator([model], cfg)
|
| 52 |
|
| 53 |
|
|
|
|
| 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 |
|