Update modeling_lang.py

modeling_lang.py

@@ -19,10 +19,6 @@ class LangDetectorModel(PreTrainedModel):
 
         # Dummy for device checking
         self.dummy_param = nn.Parameter(torch.zeros(1))
-        # Load floret model
-        print('Does it arrive here:', self.config.config)
-
-                # Get filename from config
         bin_filename = self.config.config.filename
 
         # Check if the file is already present locally, else download it
@@ -30,17 +26,10 @@ class LangDetectorModel(PreTrainedModel):
             # print(f"{bin_filename} not found locally, downloading from Hugging Face hub...")
             bin_filename = hf_hub_download(repo_id=self.config.config._name_or_path, 
                                        filename=bin_filename)
-            # print(f"Downloaded .bin file at: {bin_filename}")
-        # else:
-            # print(f"{bin_filename} found locally.")
 
         # Load floret model using the full path
         self.model_floret = floret.load_model(bin_filename)
 
-        
-        # self.model_floret = floret.load_model(self.config.config.filename)
-
-    #
     def forward(self, input_ids, **kwargs):
         if isinstance(input_ids, str):
             # If the input is a single string, make it a list for floret