FullGeneration.py 246 B

12345678
  1. import numpy as np
  2. from typing import Any, TypedDict
  3. class FullGeneration(TypedDict):
  4. semantic_prompt: np.ndarray
  5. coarse_prompt: np.ndarray # shape (2, N), type int16/int64
  6. fine_prompt: np.ndarray # shape (8, N), type int16/int64