minor changes

1. added .so file to the ignored list
2. removed 2 unused imports from test_inference.py, which also
   avoided a warning for me that was produced by importing
   pandas
This commit is contained in:
Min Xu
2024-02-02 20:19:02 -08:00
parent c0ddebaaaf
commit 8e13598868
2 changed files with 4 additions and 2 deletions

View File

@@ -18,7 +18,6 @@ from exllamav2.mlp import ExLlamaV2MLP
from exllamav2.moe_mlp import ExLlamaV2MoEMLP
import argparse, os, math, time
import pandas, fastparquet
import torch
import torch.nn.functional as F
from conversion.tokenize import get_tokens