Skip to content

Commit

Permalink
make doc
Browse files Browse the repository at this point in the history
  • Loading branch information
tsachiblau committed Nov 8, 2024
1 parent 70067d8 commit 9397314
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/peft/tuners/cpt/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,8 +24,8 @@

class CPTEmbedding(torch.nn.Module):
"""
CPTEmbedding is a custom embedding layer designed for Context-aware Prompt Tuning (CPT) in PEFT.
It initializes embeddings, applies prompt-specific projections, and computes loss using label masks.
CPTEmbedding is a custom embedding layer designed for Context-aware Prompt Tuning (CPT) in PEFT. It initializes
embeddings, applies prompt-specific projections, and computes loss using label masks.
"""

def __init__(self, config, word_embeddings):
Expand Down

0 comments on commit 9397314

Please sign in to comment.