"src/diffusers/quantizers/pipe_quant_config.py" did not exist on "eadf0e2555cfa19b033e02de53553f71ac33536f"
Unverified Commit 101d2ae9 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Rename has_origin_eids to has_original_eids to keep consistency. (#6446)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent d2eca855
...@@ -498,12 +498,17 @@ class CSCSamplingGraph(SamplingGraph): ...@@ -498,12 +498,17 @@ class CSCSamplingGraph(SamplingGraph):
""" """
# Ensure nodes is 1-D tensor. # Ensure nodes is 1-D tensor.
self._check_sampler_arguments(nodes, fanouts, probs_name) self._check_sampler_arguments(nodes, fanouts, probs_name)
has_origin_eids = ( has_original_eids = (
self.edge_attributes is not None self.edge_attributes is not None
and ORIGINAL_EDGE_ID in self.edge_attributes and ORIGINAL_EDGE_ID in self.edge_attributes
) )
return self._c_csc_graph.sample_neighbors( return self._c_csc_graph.sample_neighbors(
nodes, fanouts.tolist(), replace, False, has_origin_eids, probs_name nodes,
fanouts.tolist(),
replace,
False,
has_original_eids,
probs_name,
) )
def sample_layer_neighbors( def sample_layer_neighbors(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment