phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

chore: remove straggler references to llama-models (#1345)

Browse source 
Straggler references cleanup
This commit is contained in:
Ashwin Bharambe 2025-03-01 14:26:03 -08:00 • committed by GitHub
parent 8bbd52bb9f
commit 46b0a404e8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
19 changed files with 827 additions and 74 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/models/llama/llama3/model.py
View file

@ -27,11 +27,7 @@ from fairscale.nn.model_parallel.layers import (
)
from torch import nn
from ..api import ModelArgs
# **NOTE**: This code is not runnable without installing `torch` and `fairscale`
# dependencies. These dependencies are not part of the default dependencies
# (requirements.txt) of the `llama-models` package.
from .args import ModelArgs
class RMSNorm(torch.nn.Module):