Jiahuan's picture
Update README.md
e55818e verified
metadata
license: apache-2.0
task_categories:
  - text-generation
language:
  - en

Model Card for Model ID

Model Details

Model Description

voxreality-arta-lego-llama2-7b-chat is a Large Language Model (LLM) that is a fine-tuned version of Llama-2-7b-chat-hf using LEGO assembly dataset. This model aims to act as a training assistant for assembly task in an extended reality application.

How to use the Model

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("voxreality/voxreality-arta-lego-llama2-7b-chat", device_map="auto", trust_remote_code=True, token=True)
model = AutoModelForCausalLM.from_pretrained("voxreality/voxreality-arta-lego-llama2-7b-chat", device_map="auto", trust_remote_code=True, token=True, quantization_config=bnb_config)

More details

  • Paper: Autonomous Workflow for Multimodal Fine-Grained Training Assistants Towards Mixed Reality, ACL 2024