Upload modeling_hf.py with huggingface_hub

modeling_hf.py

@@ -307,7 +307,7 @@ class StreamMixerModel(StreamMixerPreTrainedModel):
         x = self.inner.ln0(self.inner.wte(input_ids))
         for block in self.inner.blocks:
             x = block(x)
-        return x
+        return x.float()
 
     def forward(
         self,