from models.llama3_1 --> from llama_models.llama3_1

This commit is contained in:
Hardik Shah 2024-07-21 19:07:02 -07:00
parent c6ef16f6bd
commit c64b8cba22
22 changed files with 29 additions and 27 deletions

View file

@ -7,7 +7,7 @@ from typing import Optional
import torch
from fairscale.nn.model_parallel.mappings import reduce_from_model_parallel_region
from models.llama3_1.api.model import Transformer, TransformerBlock
from llama_models.llama3_1.api.model import Transformer, TransformerBlock
from termcolor import cprint