How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="AI4free/Jarvis-0.5B")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("AI4free/Jarvis-0.5B")
model = AutoModelForCausalLM.from_pretrained("AI4free/Jarvis-0.5B", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

Overview

Jarvis-0.5B is a text generation model, Inspired by the fictional AI assistant Jarvis from the Iron Man series, this model aims to emulate Jarvis's conversational abilities. With a total of 0.5 billion parameters, Jarvis-0.5B is designed to handle various natural language understanding and generation tasks.

Model Details

  • Model Name: Jarvis-0.5B
  • Authors: OEvortex
  • Parameters: 0.5 billion
  • Architecture: Transformers

Intended Use

Jarvis-0.5B is intended for tasks requiring text generation, conversational interfaces, and natural language understanding. It can be used in various applications such as chatbots, virtual assistants, and dialogue systems.

Downloads last month
92
Safetensors
Model size
0.5B params
Tensor type
F16
·
Inference Providers NEW
Input a message to start chatting with AI4free/Jarvis-0.5B.

Model tree for AI4free/Jarvis-0.5B

Quantizations
1 model

Dataset used to train AI4free/Jarvis-0.5B