Downloads · 30 days
0
namanbnsl/moose-mini
moose-mini is a machine learning model from namanbnsl. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
0
Access
Public
Updated Jul 9, 2024
Repo size
710 MB
Likes
1
Public
Click a slice to open those files.
.pth710 MB · 100%
From the Hugging Face model README
import torch
from huggingface_hub import hf_hub_download
moose = hf_hub_download(repo_id="namanbnsl/moose-mini", filename="model.py")
weights = hf_hub_download(repo_id="namanbnsl/moose-mini", filename="model.pth")
exec(open(moose).read())
params = ModelArgs()
model = Moose(params)
model.load_state_dict(torch.load(weights))
model.to(params.device)
print(model.generate("Once upon a time, there was a little car named Beep."))