Update rita_modeling.py

rita_modeling.py

@@ -246,4 +246,4 @@ class RITAModel(PreTrainedModel):
         return self.projector
 
     def set_output_embeddings(self, new_projector):
-        return new_projector
+        self.projector = new_projector