SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Refactor: Removed un-necessary object base class (#32230)

Browse Source 
* Refactored to remove un-necessary object base class.

* small fix.
This commit is contained in:
Sai-Suraj-27
2024-07-26 14:03:02 +05:30
committed by GitHub
parent 1c7ebf1d6e
commit b8e5cd5396
56 changed files with 89 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/research_projects/performer/modeling_flax_performer_utils.py
View File

@@ -284,7 +284,7 @@ def make_fast_generalized_attention(
return attention_fn
class RandomMatrix(object):
class RandomMatrix:
r"""
Abstract class providing a method for constructing 2D random arrays. Class is responsible for constructing 2D
random arrays.
@@ -348,7 +348,7 @@ class GaussianOrthogonalRandomMatrix(RandomMatrix):
return jnp.matmul(jnp.diag(multiplier), final_matrix)
class FastAttention(object):
class FastAttention:
r"""
Abstract class providing a method for fast attention. Class is responsible for providing a method
<dot_product_attention> for fast approximate attention.