 \
  --lr_warmup_steps=0 \
  --num_class_images=200 \
  --max_train_steps=800

8 GB GPU
	
For 8GB GPUs, you’ll need the help of DeepSpeed to offload some
tensors from the VRAM to either the CPU or NVME, enabling training with less GPU memory.
Run the following command to configure your 🤗 Accelerate environment:


	Copied
accelerate config
During configuration, confirm that you want to use DeepSpeed. Now it’s possible to train on under 8GB VRAM 