Skip to content

Commit 88adf4c

Browse files
committed
refine
1 parent f6b34c1 commit 88adf4c

1 file changed

Lines changed: 0 additions & 87 deletions

File tree

backend/bizyengine/bizyair_extras/nodes_rf_inversion.py

Lines changed: 0 additions & 87 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)