fix some bug in RVC

trellis/pipelines/trellis_image_to_3d.py

@@ -643,11 +643,15 @@ class TrellisImageTo3DPipeline(Pipeline):
             feats=torch.randn(coords.shape[0], flow_model.in_channels).to(self.device),
             coords=coords,
         )
+        std = torch.tensor(self.slat_normalization['std'])[None].to(self.device)
+        mean = torch.tensor(self.slat_normalization['mean'])[None].to(self.device)
         sampler_params = {**self.slat_sampler_params, **sampler_params}
         slat = self.slat_sampler.sample_slat_opt_delta_v(
             flow_model,
             slat_decoder_gs,
             slat_decoder_mesh,
+            std,
+            mean,
             self.dreamsim_model,
             apperance_learning_rate,
             start_t,
@@ -660,8 +664,6 @@ class TrellisImageTo3DPipeline(Pipeline):
             verbose=True
         ).samples
 
-        std = torch.tensor(self.slat_normalization['std'])[None].to(slat.device)
-        mean = torch.tensor(self.slat_normalization['mean'])[None].to(slat.device)
         slat = slat * std + mean
         # from trellis.utils import render_utils, postprocessing_utils
         # import imageio