The AMD Llama 350M Upgraded is a transformer-based causal language model built on the Llama architecture, designed to generate human-like text. This model has been upgraded from the original AMD Llama 135M model to provide enhanced performance with an increased parameter count of 332 million. It is suitable for various natural language processing tasks, including text generation, completion, and conversational applications.

Model Details

Model Type: Causal Language Model
Architecture: Llama
Number of Parameters: 332 million
Input Size: Variable-length input sequences
Output Size: Variable-length output sequences

Usage

To use the AMD Llama 350M Upgraded model, you can utilize the transformers library. Here’s a sample code snippet to get started:

python

import torch
from transformers import LlamaForCausalLM, LlamaTokenizer

# Load the tokenizer and model
model_name = "reflex-ai/AMD-Llama-350M-Upgraded"
tokenizer = LlamaTokenizer.from_pretrained(model_name)
model = LlamaForCausalLM.from_pretrained(model_name)

# Set the model to evaluation mode
model.eval()

# Function to generate text
def generate_text(prompt, max_length=50):
    inputs = tokenizer.encode(prompt, return_tensors='pt', padding=True, truncation=True)
    attention_mask = (inputs != tokenizer.pad_token_id).long()

    if torch.cuda.is_available():
        inputs = inputs.to('cuda')
        attention_mask = attention_mask.to('cuda')

    with torch.no_grad():
        outputs = model.generate(inputs, attention_mask=attention_mask, max_length=max_length, num_return_sequences=1)
    
    generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
    return generated_text

# Example usage
prompt = "Once upon a time in a land far away,"
generated_output = generate_text(prompt, max_length=100)
print(generated_output)