Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
Commit
•
fffa5a7
1
Parent(s):
b4c89cc
Update model/model_manager.py
Browse files- model/model_manager.py +1 -1
model/model_manager.py
CHANGED
@@ -55,7 +55,7 @@ class ModelManager:
|
|
55 |
def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
|
56 |
pipe = self.load_model_pipe(model_name)
|
57 |
if 'PNP' in model_name:
|
58 |
-
result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inversion_steps=
|
59 |
elif 'Prompt2prompt' in model_name:
|
60 |
result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inner_steps=5)
|
61 |
else:
|
|
|
55 |
def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
|
56 |
pipe = self.load_model_pipe(model_name)
|
57 |
if 'PNP' in model_name:
|
58 |
+
result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inversion_steps=100)
|
59 |
elif 'Prompt2prompt' in model_name:
|
60 |
result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inner_steps=5)
|
61 |
else:
|