Fix: Rename shift parameter to flow_shift in UniPCMultistepScheduler initialization across SkyReels test files

tolgacangoz · tolgacangoz · commit fc3d3282e002 · 2025-07-03T17:15:01.000+03:00
diff --git a/tests/pipelines/skyreels_v2/test_skyreels_v2.py b/tests/pipelines/skyreels_v2/test_skyreels_v2.py
@@ -67,7 +67,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=8.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=8.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 
diff --git a/tests/pipelines/skyreels_v2/test_skyreels_v2_df.py b/tests/pipelines/skyreels_v2/test_skyreels_v2_df.py
@@ -67,7 +67,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=8.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=8.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 
diff --git a/tests/pipelines/skyreels_v2/test_skyreels_v2_df_image_to_video.py b/tests/pipelines/skyreels_v2/test_skyreels_v2_df_image_to_video.py
@@ -67,7 +67,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 
@@ -158,7 +158,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 
diff --git a/tests/pipelines/skyreels_v2/test_skyreels_v2_df_video_to_video.py b/tests/pipelines/skyreels_v2/test_skyreels_v2_df_video_to_video.py
@@ -69,7 +69,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 
diff --git a/tests/pipelines/skyreels_v2/test_skyreels_v2_image_to_video.py b/tests/pipelines/skyreels_v2/test_skyreels_v2_image_to_video.py
@@ -70,7 +70,7 @@ def get_dummy_components(self):
         )
 
         torch.manual_seed(0)
-        scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)
+        scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)
         text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
         tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")
 

Original file line number	Diff line number	Diff line change
`@@ -67,7 +67,7 @@ def get_dummy_components(self):`
`67`	`67`	`)`
`68`	`68`
`69`	`69`	`torch.manual_seed(0)`
`70`		`- scheduler = UniPCMultistepScheduler(shift=8.0, use_flow_sigmas=True)`
	`70`	`+ scheduler = UniPCMultistepScheduler(flow_shift=8.0, use_flow_sigmas=True)`
`71`	`71`	`text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")`
`72`	`72`	`tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")`
`73`	`73`
Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def get_dummy_components(self):`
`69`	`69`	`)`
`70`	`70`
`71`	`71`	`torch.manual_seed(0)`
`72`		`- scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)`
	`72`	`+ scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)`
`73`	`73`	`text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")`
`74`	`74`	`tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")`
`75`	`75`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def get_dummy_components(self):`
`70`	`70`	`)`
`71`	`71`
`72`	`72`	`torch.manual_seed(0)`
`73`		`- scheduler = UniPCMultistepScheduler(shift=5.0, use_flow_sigmas=True)`
	`73`	`+ scheduler = UniPCMultistepScheduler(flow_shift=5.0, use_flow_sigmas=True)`
`74`	`74`	`text_encoder = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")`
`75`	`75`	`tokenizer = AutoTokenizer.from_pretrained("hf-internal-testing/tiny-random-t5")`
`76`	`76`