Update src/loss.py

src/loss.py

@@ -24,7 +24,7 @@ class LossScheduler:
 		else:A.t_prev=B
 		return C,
 class SchedulerWrapper:
-	def __init__(A,scheduler,loss_params_path='optimized_loss_params_try1.pth'):A.scheduler=scheduler;A.catch_x,A.catch_e,A.catch_x_={},{},{};A.loss_scheduler=_A;A.loss_params_path=loss_params_path
+	def __init__(A,scheduler,loss_params_path='optimized_loss_params_re_hooked_model.pth'):A.scheduler=scheduler;A.catch_x,A.catch_e,A.catch_x_={},{},{};A.loss_scheduler=_A;A.loss_params_path=loss_params_path
 	def set_timesteps(A,num_inference_steps,**C):
 		D=num_inference_steps
 		if A.loss_scheduler is _A:B=A.scheduler.set_timesteps(D,**C);A.timesteps=A.scheduler.timesteps;A.init_noise_sigma=A.scheduler.init_noise_sigma;A.order=A.scheduler.order;return B