tianleliphoebe commited on
Commit
66e660a
1 Parent(s): 006c7d1

Update model/models/imagenhub_models.py

Browse files
Files changed (1) hide show
  1. model/models/imagenhub_models.py +2 -2
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"] = 300
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"] = 3
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):