Update rita_modeling.py

rita_modeling.py

@@ -1,35 +1,3 @@
-
-Hugging Face's logo Hugging Face
-
-Models
-Datasets
-Spaces
-Docs
-Pricing
-Please check your email address for a confirmation link
-nz
-/
-RITA_s
-PyTorch
-Transformers
-rita
-Model card
-Files and versions
-Settings
-RITA_s
-/ rita_modeling.py
-nz's picture
-nz
-Update rita_modeling.py
-b98e56e
-about 21 hours ago
-raw
-history
-blame
-edit
-delete
-Safe
-8.68 kB
 import math
 import os
 from dataclasses import dataclass
@@ -278,4 +246,4 @@ class RITAModel(PreTrainedModel):
         return self.projector
 
     def set_output_embeddings(self, new_projector):
-        self.projector = new_projector
+        self.projector = new_projector