Update loader.py
This commit is contained in:
parent
a00fec24f8
commit
e79fdebbe8
1 changed files with 1 additions and 1 deletions
|
|
@ -348,7 +348,7 @@ class FastLanguageModel(FastLlamaModel):
|
|||
# dispatch_model = FastGraniteModel
|
||||
else:
|
||||
return FastModel.from_pretrained(
|
||||
model_name = model_name,
|
||||
model_name = old_model_name,
|
||||
max_seq_length = max_seq_length,
|
||||
dtype = dtype,
|
||||
load_in_4bit = load_in_4bit,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue