"megatron/inference/text_generation/generation.py" did not exist on "1d391bba132ac2cb6077ee10bc4138a7260d39f2"
  • Charlie Lin's avatar
    Dynamic ref flatten (#1482) · 4c32afcc
    Charlie Lin authored
    Changes flatten's compute_shape() to handle dynamic shapes
    Calculates the flattened shape with the min, max, and opt
    4c32afcc