Spaces:
Running
on
Zero
Running
on
Zero
tianleliphoebe
commited on
Commit
•
66e660a
1
Parent(s):
006c7d1
Update model/models/imagenhub_models.py
Browse files
model/models/imagenhub_models.py
CHANGED
@@ -13,7 +13,7 @@ class PNP(ImagenHubModel):
|
|
13 |
|
14 |
def __call__(self, *args, **kwargs):
|
15 |
if "num_inversion_steps" not in kwargs:
|
16 |
-
kwargs["num_inversion_steps"] =
|
17 |
return super().__call__(*args, **kwargs)
|
18 |
|
19 |
class Prompt2prompt(ImagenHubModel):
|
@@ -22,7 +22,7 @@ class Prompt2prompt(ImagenHubModel):
|
|
22 |
|
23 |
def __call__(self, *args, **kwargs):
|
24 |
if "num_inner_steps" not in kwargs:
|
25 |
-
kwargs["num_inner_steps"] =
|
26 |
return super().__call__(*args, **kwargs)
|
27 |
|
28 |
def load_imagenhub_model(model_name, model_type=None):
|
|
|
13 |
|
14 |
def __call__(self, *args, **kwargs):
|
15 |
if "num_inversion_steps" not in kwargs:
|
16 |
+
kwargs["num_inversion_steps"] = 200
|
17 |
return super().__call__(*args, **kwargs)
|
18 |
|
19 |
class Prompt2prompt(ImagenHubModel):
|
|
|
22 |
|
23 |
def __call__(self, *args, **kwargs):
|
24 |
if "num_inner_steps" not in kwargs:
|
25 |
+
kwargs["num_inner_steps"] = 2
|
26 |
return super().__call__(*args, **kwargs)
|
27 |
|
28 |
def load_imagenhub_model(model_name, model_type=None):
|