Henry65
/

RepoSim4Py

Feature Extraction

code-understanding

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

Henry65 commited on Jul 17, 2023

Commit

cb896d6

·

1 Parent(s): fb6f33c

Update RepoPipeline.py

Files changed (1) hide show

RepoPipeline.py +1 -1

RepoPipeline.py CHANGED Viewed

@@ -161,7 +161,7 @@ class RepoPipeline(Pipeline):
             if text_sets is None or len(text_sets) == 0 \
             else torch.zeros((1, 768), device=self.device)
-    def _forward(self, extracted_infos: List, max_length) -> List:
         model_outputs = []
         num_repos = len(extracted_infos)
         with tqdm(total=num_repos) as progress_bar:

             if text_sets is None or len(text_sets) == 0 \
             else torch.zeros((1, 768), device=self.device)
+    def _forward(self, extracted_infos: List, max_length=512) -> List:
         model_outputs = []
         num_repos = len(extracted_infos)
         with tqdm(total=num_repos) as progress_bar: