Update src/loss.py

src/loss.py

@@ -137,7 +137,7 @@ class SchedulerWrapper:
 		A.loss_scheduler=_A
 		A.loss_params_path=loss_params_path
 	def set_timesteps(A,num_inference_steps,**C):
-		D=num_inference_steps
+		D=11
 		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
 		else:B=A.loss_scheduler.set_timesteps(D,**C);A.timesteps=A.loss_scheduler.timesteps;A.init_noise_sigma=A.scheduler.init_noise_sigma;A.order=A.scheduler.order;return B
 	def step(B,model_output,timestep,sample,**F):