FLUX-1 DEV LoRA Training on Google Colab

(ต้องใช้ A100 ซึ่งต้องสมัคร Colab Pro)

ข้อดี

• เทรนเร็ว 1,000 steps ใช้เวลาประมาณ 40 นาที ต่อ 10 หน่วยประมวลผล ตก 35บาท/1,000 steps

ข้อเสีย

• ยังงง ๆ เรื่องการแท็ก

• สำหรับเทรนสไตล์อาจต้องปรับจูนเองเยอะ ต่างจาก civitai ที่เตรียมทุกอย่างให้พร้อม แต่ถ้าเทรนหน้าไม่ต้องเตรียมอะไรมาก ใช้สูตรนี้ได้เลย (หรือ จะลดเสต็ปดูก็ได้ แต่ช่วงที่ดีจะอยู่ระหว่าง 1,000 - 4,000)

• ผมเทรนลองเทรนบน A100GPU ไป 4,000 steps ใช้ไป 2-3 ชั่วโมงกว่า หากคิดกลม ๆ ก็ 1,000 steps ต่อ 40 นาที ต่อ 10 หน่วยประมวลผล

  
  

• ไปที่ลิงก์ Colab https://colab.research.google.com/drive/1r09aImgL1YhQsJgsLWnb67-bjTV88-W0

• สมัคร Colab Pro (ราว ๆ 350 บาท ต่อเดือน หรือ 9.99$) เพื่อใช้ A100 GPU

• ที่มุมขวา กดเปลี่ยนรันไทม์จาก L4 เป็น A100 จากนั้นกดรันตามเสต็ป

• ถ้าเชื่อมต่อไม่สำเร็จและยังกลับไปขึ้น L4 ให้ยกเลิกการเชื่อมต่อ > delete runtime > ลองใหม่จนกว่าจะได้

• จากนั้นสิ่งที่ต้องทำต่อคือเพิ่มภาพเข้าไปใน Dataset ใช้ประมาณ 20 ภาพขึ้นไปกำลังดี ขนาด 1024x1024

• Model License - ล็อคอินHugginface แล้วยอมรับเงื่อนไขข้อตกลงการใช้โมเดลที่นี่ black-forest-labs/FLUX.1-dev

• จากนั้นกดรับ Read Token Type Key Get a READ key from huggingface แล้ววางไว้ในช่องว่างหลังจากกดรัน จากนั้น Enter

• ปรับเปลี่ยนเซ็ตติ้งทั่วไป เช่น ชื่อ LoRa , Trigger word, Steps ตามใจชอบ ต้องอ่านเยอะหน่อยนะว่าอันไหนบอกอะไร แต่ผมเปลี่ยนแค่ชื่อกับ Triggerword ในการทดสอบครั้งแรก

• Run !

Link: https://colab.research.google.com/drive/1r09aImgL1YhQsJgsLWnb67-bjTV88-W0

Repo: https://github.com/ostris/ai-toolkit

AI Toolkit by Ostris https://ostris.com/

Special Thanks : @angrypenguinpng