LoRA-SD-training

מה זה

האפליקציה מחולקת לשלושה חלקים עיקריים שכוללים אימון, בדיקה והעלאה. בחלק של האימון אתם מעלים קובצי תמונות עבור מושג מסוים, מגדירים תיאור טקסטואלי וקובעים פרמטרים כמו מספר צעדי אימון, קצב למידה, שימוש ב־FP16 ו־8bit Adam, או אימון מקודד הטקסט ברזולוציה קבועה של 512.

הבסיס שמריץ את התהליך מאחורי הקלעים הוא stabilityai/stable-diffusion-2-1-base בטכניקת LoRA. בלשונית הבדיקה אפשר לבחור את קובצי המשקולות שנוצרו, להזין פרומפט, לקבוע פרמטרים כמו סולם CFG, מספר צעדים וערכי אלפא, ולקבל תמונה מוכנה. לשונית ההעלאה מאפשרת לשלוח את התוצאה ישירות באמצעות טוקן ייעודי.

מתאים ל

• אימון LoRA לתמונות

  מעלים תמונות של אובייקט, מגדירים צעדי למידה ומקבלים קובץ משקולות מותאם.

• בדיקת תוצאות מקומית

  טוענים את המשקולות שנוצרו ומייצרים תמונות ניסיון עם הגדרות אלפא שונות.

• שמירה ישירה לענן

  מזינים טוקן ושם מודל כדי לשלוח את המשקולות המאומנות למאגר האישי.

איפה זה נופל

הבעיה המרכזית כרגע היא שהאפליקציה תקועה על BUILD_ERROR ולכן לא עובדת בדפדפן כלל. מעבר לזה, החומרה המוגדרת היא מעבד בסיסי בלבד, מה שהופך אימון מודל כזה לבלתי מעשי לחלוטין בלי GPU. הרזולוציה נעולה על 512, ואין בממשק תצוגה מקדימה מובנית או דוגמאות מוכנות מראש לבדיקה מהירה.

איך משתמשים

בפועל אתם עוברים ללשונית האימון, מעלים את התמונות, מגדירים את הערכים ולוחצים על כפתור תחילת האימון. יש כפתור נפרד לבדיקת סטטוס האימון וקבלת קובצי המשקולות. כדי להעלות את התוצאות צריך להזין טוקן של Hugging Face. כרגע המרחב נמצא במצב שגיאת בנייה ורץ על מעבד בסיסי, כך שאי אפשר להשתמש בו ישירות ברשת בלי לתקן את הקוד ולהעביר אותו למעבד גרפי מתאים.

הרישיון

הפרויקט מפורסם תחת רישיון MIT, שמאפשר שימוש חופשי בקוד, שינוי והפצה שלו. החומרים שאתם מעלים והמשקולות שנוצרות כפופים לרישוי ולמגבלות השימוש של מודל הבסיס Stable Diffusion 2.1.

• שימוש מסחרי
• שינוי הקוד
• הפצה מחדש

הרישיון המלא ב־Hugging Face ↗