Update modeling_quiet.py

modeling_quiet.py

@@ -1246,7 +1246,7 @@ class QuietForCausalLM(QuietPreTrainedModel):
 
         self.policy_loss_beta = 1e6
         self.embedding_scale = 1e2
-        self.temperature = nn.Parameter(torch.tensor(1.0, size=(1,)))
+        self.temperature = nn.Parameter(torch.ones(1))
         self.max_temperature = config.max_temperature
         self.complexity_factor = config.complexity_factor
         self.reinforce_temperature = 3