Removing unused peft dependency
Browse filesPeft doesn't appear to be used by the code. Unless I'm mistaken, removing its import would not affect functionality, and would reduce the number of exterior packages needed to run.
- modeling_nvmmembed.py +0 -1
modeling_nvmmembed.py
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
import torch
|
| 2 |
import torch.nn.functional as F
|
| 3 |
-
from peft import PeftModel
|
| 4 |
from transformers import AutoTokenizer, AutoModel
|
| 5 |
|
| 6 |
|
|
|
|
| 1 |
import torch
|
| 2 |
import torch.nn.functional as F
|
|
|
|
| 3 |
from transformers import AutoTokenizer, AutoModel
|
| 4 |
|
| 5 |
|