Update src/loss.py

src/loss.py

@@ -46,7 +46,7 @@ class SchedulerWrapper:
 	def prepare_loss(A,num_accelerate_steps=15):A.load_loss_params()
 def get_instance(device):
     compress = output.Decoder().to(device).requires_grad_(False)
-    compress.load_state_dict(torch.load("taesd3_decoder.pth", map_location=device, weights_only=True)) # taesdxl_decoder.pth
+    compress.load_state_dict(torch.load("taesdxl_decoder.pth", map_location=device, weights_only=True)) # taesdxl_decoder.pth
     return compress
 @torch.no_grad()
 def hook_pipe(pipe, compress, mul, sub, scaling_factor):