@@ -154,90 +154,3 @@ def INPUT_TYPES(s):
154154 ),
155155 }
156156 }
157-
158-
159- class BizyAir_QuantizeDualCLIPLoaderOffline (BizyAirBaseNode ):
160- @classmethod
161- def INPUT_TYPES (s ):
162- return {
163- "required" : {
164- "clip_name1" : (folder_paths .get_filename_list ("clip" ),),
165- "clip_name2" : (folder_paths .get_filename_list ("clip" ),),
166- "type" : (["flux" ],),
167- }
168- }
169-
170- RETURN_TYPES = (data_types .CLIP ,)
171- FUNCTION = "load_clip"
172-
173- CATEGORY = "advanced/loaders"
174-
175- def load_clip (self , clip_name1 , clip_name2 , type ):
176-
177- node_data = create_node_data (
178- class_type = "QuantizeDualCLIPLoaderOffline" ,
179- inputs = {
180- "clip_name1" : clip_name1 ,
181- "clip_name2" : clip_name2 ,
182- "type" : type ,
183- },
184- outputs = {"slot_index" : 0 },
185- )
186- model = BizyAirNodeIO (
187- self .assigned_id ,
188- {self .assigned_id : node_data },
189- # config_file=folder_paths.guess_config(clip_name=clip_name1),
190- )
191- return (model ,)
192-
193-
194- class OneDiffLoadOptimizedDiffusionModel (BizyAirBaseNode ):
195- @classmethod
196- def INPUT_TYPES (s ):
197- return {
198- "required" : {
199- "unet_name" : (folder_paths .get_filename_list ("diffusion_models" ),),
200- "max_num_loras" : (
201- "INT" ,
202- {
203- "default" : 3 ,
204- "step" : 1 ,
205- "min" : 0 ,
206- "max" : 4 ,
207- "display" : "number" ,
208- },
209- ),
210- "max_ranks" : (
211- "INT" ,
212- {
213- "default" : 32 ,
214- "step" : 32 ,
215- "min" : 32 ,
216- "max" : 128 ,
217- "display" : "number" ,
218- },
219- ),
220- }
221- }
222-
223- RETURN_TYPES = (data_types .MODEL , "LORA_MANAGER" )
224- FUNCTION = "load_unet"
225- CATEGORY = "☁️BizyAir/OneDiffEnterprise"
226-
227- def load_unet (self , unet_name , max_num_loras , max_ranks ):
228- print ("why : " , folder_paths .get_filename_list ("diffusion_models" ))
229- node_data = create_node_data (
230- class_type = "OneDiffLoadOptimizedDiffusionModel" ,
231- inputs = {
232- "unet_name" : unet_name ,
233- "max_num_loras" : max_num_loras ,
234- "max_ranks" : max_ranks ,
235- },
236- outputs = {"slot_index" : 0 },
237- )
238-
239- model = BizyAirNodeIO (
240- self .assigned_id ,
241- {self .assigned_id : node_data },
242- )
243- return (model ,)
0 commit comments