Spaces:
Build error
Build error
Himank Jain commited on
Commit ·
39e3e15
1
Parent(s): c6a12ae
fixed tokenizer imports
Browse files- models/multimodel_phi.py +1 -1
models/multimodel_phi.py
CHANGED
|
@@ -2,7 +2,7 @@ import torch
|
|
| 2 |
import torch.nn as nn
|
| 3 |
import os
|
| 4 |
from peft import PeftModel
|
| 5 |
-
from transformers import PreTrainedModel, AutoModelForCausalLM
|
| 6 |
from .projection_layer import ProjectionBlock
|
| 7 |
|
| 8 |
class MultimodalPhiModel(PreTrainedModel):
|
|
|
|
| 2 |
import torch.nn as nn
|
| 3 |
import os
|
| 4 |
from peft import PeftModel
|
| 5 |
+
from transformers import PreTrainedModel, AutoModelForCausalLM, AutoTokenizer
|
| 6 |
from .projection_layer import ProjectionBlock
|
| 7 |
|
| 8 |
class MultimodalPhiModel(PreTrainedModel):
|