Saya butuh bantuan training data AI sampai 5000 Epoch, Berikut adalah Berikut adalah link tutorial dan source code yang saya gunakan. Sebelum bid bisa docoba dulu untuk memastikan apakah programnya sudah jalan atau belum di PC yang digunakan untuk training data.
link tutorial https://huggingface.co/docs/diffusers/v0.13.0/en/training/text2image
- download requirement.txt di https://github.com/huggingface/diffusers/tree/main/examples/text_to_image
- download train_text_to_image.py di https://github.com/huggingface/diffusers/tree/main/examples/text_to_image
pip install git+https://github.com/huggingface/diffusers.git
pip install -U -r requirements.txt
pip uninstall torch torchvision
pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
accelerate config
huggingface-cli login --token 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
Dan berikut adalah coding untuk run di CMD
set MODEL_NAME=runwayml/stable-diffusion-v1-5
set dataset_name=njoeljack/hai-sd
accelerate launch C:\Users\Dedek\Downloads\testsd\train_text_to_image.py ^
- pretrained_model_name_or_path=%MODEL_NAME% ^
- dataset_name=%dataset_name% ^
- use_ema ^
- resolution=512 ^
- random_flip ^
- train_batch_size=1 ^
- gradient_accumulation_steps=4 ^
- gradient_checkpointing ^
- mixed_precision="fp16" ^
- max_train_steps=5000 ^
- learning_rate=1e-05 ^
- max_grad_norm=1 ^
- lr_scheduler="constant" ^
- lr_warmup_steps=0 ^
- output_dir="sd-test-modelv4"
Silahkan bisa dulu, jika sudah bisa running bisa infokan ke saya untuk saya pilih bid nya
Loading ...
