GPT
O

openvla-7b

קוד פתוח

openvlamit2024-06-10

  • transformers
  • safetensors
  • openvla
  • feature-extraction
  • robotics

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

  • 7.5Bפרמטרים
  • 14.0 GBמשקל הקבצים
  • 441,512הורדות בחודש
  • 254לייקים

מה זה

המודל פותח על ידי חוקרים מסטנפורד, ברקלי, גוגל דיפמיינד וטויוטה. הוא משלב מודל שפה של Llama-2 יחד עם רשתות ראייה מסוג DINOv2 ו־SigLIP, ואומן על 970 אלף הדגמות של מניפולציות רובוטיות ממאגר Open X-Embodiment.

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

מתאים ל

  • הפעלת זרוע Widow-X

    שליטה ישירה ללא אימון נוסף בסביבות BridgeV2 שנכללו במאגר המקורי.

  • אימון משימות מותאמות

    בסיס להתאמה יעילה על מעט הדגמות של זרוע חדשה במקום להתחיל מודל מאפס.

  • מחקר רובוטיקה מעשי

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

איפה זה נופל

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

שאלות
נפוצות

האם אפשר לשלוט ברובוט עם הוראות בעברית?

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

האם הפלט של המודל נשלח ישירות למנועים?

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

האם הוא יעבוד עם כל זרוע רובוטית שיש לי במעבדה?

הוא עובד מיד רק על רובוטים וסביבות שהיו חלק ממאגר Open X-Embodiment. עבור כל חומרה אחרת, תצטרכו לאסוף סט הדגמות ולבצע fine-tuning בעזרת הסקריפטים שהחוקרים סיפקו.

איך מריצים

כדי להריץ אותו מקומית צריך לפרוק 14 גיגה-בייט של משקולות בפורמט bfloat16 ישירות לתוך הזיכרון של כרטיס המסך. בפועל, כרטיס עם 16 או 24 גיגה-בייט של VRAM הוא דרישת סף, במיוחד אם רוצים לקבל זמני תגובה שמתאימים לשליטה בזמן אמת ברובוט.

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

from transformers import pipeline

pipe = pipeline("robotics", model="openvla/openvla-7b")
print(pipe("שלום"))

הרישיון

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

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

הרישיון המלא בעמוד המודל ↗