mirror of
https://github.com/exo-explore/exo.git
synced 2025-10-23 02:57:14 +03:00
Merge pull request #506 from exo-explore/dashsounds
add missing top_p_sampling import
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import numpy as np
|
||||
import mlx.core as mx
|
||||
import mlx.nn as nn
|
||||
from mlx_lm.sample_utils import top_p_sampling
|
||||
from ..inference_engine import InferenceEngine
|
||||
from .stateful_model import StatefulModel
|
||||
from .sharded_utils import load_shard
|
||||
|
||||
Reference in New Issue
Block a user