dbaranchuk 3 anni fa
parent
commit
483b1c665f
1 ha cambiato i file con 1 aggiunte e 2 eliminazioni
  1. 1 2
      src/client/remote_model.py

+ 1 - 2
src/client/remote_model.py

@@ -192,7 +192,6 @@ class DistributedBloomForSequenceClassification(BloomForSequenceClassification):
 
 
         self.transformer = DistributedBloomModel(config)
         self.transformer = DistributedBloomModel(config)
         self.score = nn.Linear(config.hidden_size, config.num_labels, bias=False)
         self.score = nn.Linear(config.hidden_size, config.num_labels, bias=False)
-        
+
         # Initialize weights and apply final processing
         # Initialize weights and apply final processing
         self.post_init()
         self.post_init()
-