python linter
This commit is contained in:
parent
4be934c453
commit
7385f7d7e2
1 changed files with 1 additions and 0 deletions
|
@ -2561,6 +2561,7 @@ class Phi3MiniModel(Model):
|
||||||
yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_LONG), torch.tensor(long_factors, dtype=torch.float32))
|
yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_LONG), torch.tensor(long_factors, dtype=torch.float32))
|
||||||
yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_SHORT), torch.tensor(short_factors, dtype=torch.float32))
|
yield (self.format_tensor_name(gguf.MODEL_TENSOR.ROPE_FACTORS_SHORT), torch.tensor(short_factors, dtype=torch.float32))
|
||||||
|
|
||||||
|
|
||||||
@Model.register("PhiMoEForCausalLM")
|
@Model.register("PhiMoEForCausalLM")
|
||||||
class PhiMoeModel(Phi3MiniModel):
|
class PhiMoeModel(Phi3MiniModel):
|
||||||
model_arch = gguf.MODEL_ARCH.PHIMOE
|
model_arch = gguf.MODEL_ARCH.PHIMOE
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue