Commit
·
ca09939
1
Parent(s):
448868f
add model
Browse files- rita_modeling.py +0 -1
rita_modeling.py
CHANGED
|
@@ -240,7 +240,6 @@ class RITAModel(PreTrainedModel):
|
|
| 240 |
output_hidden_states=None,
|
| 241 |
return_dict=None) -> torch.FloatTensor:
|
| 242 |
|
| 243 |
-
print("hello?")
|
| 244 |
if inputs_embeds == None:
|
| 245 |
x = self.embedding(input_ids) # N x L x D
|
| 246 |
else:
|
|
|
|
| 240 |
output_hidden_states=None,
|
| 241 |
return_dict=None) -> torch.FloatTensor:
|
| 242 |
|
|
|
|
| 243 |
if inputs_embeds == None:
|
| 244 |
x = self.embedding(input_ids) # N x L x D
|
| 245 |
else:
|