From 280441d571247e681d60b2edff5b5286c653a040 Mon Sep 17 00:00:00 2001 From: Felix Terkhorn Date: Tue, 1 Nov 2022 16:48:59 -0400 Subject: [PATCH] fix typo specifying number of GPUs --- stable-diffusion-finetuning/pokemon_finetune.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable-diffusion-finetuning/pokemon_finetune.ipynb b/stable-diffusion-finetuning/pokemon_finetune.ipynb index 79aa765..dd3612c 100644 --- a/stable-diffusion-finetuning/pokemon_finetune.ipynb +++ b/stable-diffusion-finetuning/pokemon_finetune.ipynb @@ -142,7 +142,7 @@ " --finetune_from \"$ckpt_path\" \\\n", " data.params.batch_size=\"$BATCH_SIZE\" \\\n", " lightning.trainer.accumulate_grad_batches=\"$ACCUMULATE_BATCHES\" \\\n", - " data.params.validation.params.n_gpus=\"$NUM_GPUS\" \\\n", + " data.params.validation.params.n_gpus=\"$N_GPUS\" \\\n", ")" ] },