oiNSSKJr SSKrSSKJr SSKrSSKJr "SS\5rg)) annotationsN) OrderedDict)Modulec^\rSrSrSrSSU4SjjjrSrSrSrSr Sr S r S r S r S rS rSrSrSrSrSrU=r$) BufferDicta Holds buffers in a dictionary. BufferDict can be indexed like a regular Python dictionary, but buffers it contains are properly registered, and will be visible by all Module methods. `torch.nn.BufferDict` is an **ordered** dictionary that respects * the order of insertion, and * in `torch.nn.BufferDict.update`, the order of the merged `OrderedDict` or another `torch.nn.BufferDict` (the argument to `torch.nn.BufferDict.update`). Note that `torch.nn.BufferDict.update` with other unordered mapping types (e.g., Python's plain `dict`) does not preserve the order of the merged mapping. Args: buffers (iterable, optional): a mapping (dictionary) of (string : `torch.Tensor`) or an iterable of key-value pairs of type (string, `torch.Tensor`) ```python class MyModule(nn.Module): def __init__(self): super().__init__() self.buffers = nn.BufferDict({"left": torch.randn(5, 10), "right": torch.randn(5, 10)}) def forward(self, x, choice): x = self.buffers[choice].mm(x) return x ``` cX>[TU]5 X lUbURU5 gg)z Args: buffers (`dict`): A mapping (dictionary) from string to `torch.Tensor`, or an iterable of key-value pairs of type (string, `torch.Tensor`). N)super__init__ persistentupdate)selfbuffersr __class__s R/home/james-whalen/.local/lib/python3.13/site-packages/peft/tuners/_buffer_dict.pyr BufferDict.__init__2s, $   KK  c URU$N_buffersrkeys r __getitem__BufferDict.__getitem__?s}}S!!rc8URXURS9 g)Nr )register_bufferr )rrbuffers r __setitem__BufferDict.__setitem__Bs ST__ErcURU grrrs r __delitem__BufferDict.__delitem__Es MM# rc,[UR5$r)lenrrs r__len__BufferDict.__len__Hs4==!!rcH[URR55$r)iterrkeysr's r__iter__BufferDict.__iter__KsDMM&&())rcXR;$rrrs r __contains__BufferDict.__contains__Nsmm##rc8URR5 g)z%Remove all items from the BufferDict.N)rclearr's rr3BufferDict.clearQs rcXnX U$)zuRemove key from the BufferDict and return its buffer. Args: key (`str`): Key to pop from the BufferDict )rrvs rpopBufferDict.popUs I Irc6URR5$)z*Return an iterable of the BufferDict keys.)rr,r's rr,BufferDict.keys`s}}!!##rc6URR5$)z5Return an iterable of the BufferDict key/value pairs.)ritemsr's rr=BufferDict.itemsds}}""$$rc6URR5$)z,Return an iterable of the BufferDict values.)rvaluesr's rr@BufferDict.valueshs}}##%%rc [U[RR5(d![ S[ U5R -5e[U[[45(aUR5H up#X0U'M g [U[RR5(a'[UR55H up#X0U'M g [U5HupE[U[RR5(d0[ S[U5-S-[ U5R -5e[U5S:Xd2[S[U5-S-[[U55-S-5eUSXS'M g ) a Update the `torch.nn.BufferDict` with the key-value pairs from a mapping or an iterable, overwriting existing keys. Note: If `buffers` is an `OrderedDict`, a `torch.nn.BufferDict`, or an iterable of key-value pairs, the order of new elements in it is preserved. Args: buffers (iterable): a mapping (dictionary) from string to `torch.Tensor`, or an iterable of key-value pairs of type (string, `torch.Tensor`). zQBuffersDict.update should be called with an iterable of key/value pairs, but got z$BufferDict update sequence element #z should be Iterable; isz has length z; 2 is requiredrN) isinstance collectionsabcIterable TypeErrortype__name__rrr=Mappingsorted enumeratestrr& ValueError)rrrrjps rr BufferDict.updatels[';??#;#;<<8:>w-:P:PQ  g Z8 9 9&}} "S  / !8!8 9 9%gmmo6 "S  7"'*!![__%=%=>>#>QGJccfjklfmfvfvv1v{$!!f%'568;CF DFWXqTqT +rc/nURR5Hup#SRSUR555nURR nUS:XaSO$SUR 5SUR5S3nS[R"U5S UUS 3nURS U-S -U-5 M S RU5nU$)Nxc38# UHn[U5v M g7fr)rO).0sizes r (BufferDict.extra_repr..s?hdD hscpuz ( )zBuffer containing: [z of size ]z (z):  ) rr=joinrXdevicerJupper get_devicetorchtypenameappend) r child_lineskrRsize_str device_type device_strparastrtmpstrs r extra_reprBufferDict.extra_reprs MM'')DAxx?affh??H((--K*e32k>O>O>Q=RRSTUT`T`TbSccd9eJ,U^^A->,?y S]R^^_`G   uqy507: ; * ;' rc[S5e)Nz BufferDict should not be called.) RuntimeError)rinputs r__call__BufferDict.__call__s=>>rr)NF)r bool)rK __module__ __qualname____firstlineno____doc__r rr r#r(r-r0r3r8r,r=r@r rort__static_attributes__ __classcell__)rs@rrrs_< ! !"F"*$ $%&%"N ??rr) __future__rrFrretorch.nnrrr6rrrs$## L?L?r