update replicate

main
Brett Kuprel 2 years ago
parent 355ef19157
commit f875572a9b
  1. 24
      min_dalle.ipynb
  2. 22
      replicate_predictor.py

24
min_dalle.ipynb vendored

File diff suppressed because one or more lines are too long

@ -27,22 +27,22 @@ class ReplicatePredictor(BasePredictor):
le=8,
default=4
),
log2_supercondition_factor: int = Input(
description='Higher values result in better agreement with the text but a narrower variety of generated images',
ge=1,
le=6,
default=4
log2_temperature: float = Input(
description='A higher temperature results in more variety.',
ge=-3,
le=3,
default=0
),
) -> Iterator[Path]:
try:
seed = -1
log2_mid_count = 3 if intermediate_outputs else 0
image_stream = self.model.generate_image_stream(
text,
seed,
text = text,
seed = -1,
grid_size = grid_size,
log2_mid_count=log2_mid_count,
log2_supercondition_factor=log2_supercondition_factor,
log2_mid_count = 3 if intermediate_outputs else 0,
temperature = 2 ** log2_temperature,
supercondition_factor = 2 ** 4,
top_k = 2 ** 8,
is_verbose = True
)

Loading…
Cancel
Save