biXNddlmZddlZddlmZddlZddlmZGddeZy)) annotationsN) OrderedDict)ModuleczeZdZdZddfd ZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZxZS) BufferDictaq 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 ``` cXt|||_||j|yy)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 S/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/peft/tuners/_buffer_dict.pyr zBufferDict.__init__2s. $   KK  c |j|SN_buffersr keys r __getitem__zBufferDict.__getitem__?s}}S!!rc@|j|||jy)N)r )register_bufferr )r rbuffers r __setitem__zBufferDict.__setitem__Bs S&T__Erc|j|=yrrrs r __delitem__zBufferDict.__delitem__Es MM# rc,t|jSr)lenrr s r__len__zBufferDict.__len__Hs4==!!rcHt|jjSr)iterrkeysr!s r__iter__zBufferDict.__iter__KsDMM&&())rc||jvSrrrs r __contains__zBufferDict.__contains__Nsdmm##rc8|jjy)z%Remove all items from the BufferDict.N)rclearr!s rr*zBufferDict.clearQs rc||}||=|S)zRemove key from the BufferDict and return its buffer. Args: key (`str`): Key to pop from the BufferDict )r rvs rpopzBufferDict.popUs I Irc6|jjS)z*Return an iterable of the BufferDict keys.)rr%r!s rr%zBufferDict.keys`s}}!!##rc6|jjS)z5Return an iterable of the BufferDict key/value pairs.)ritemsr!s rr1zBufferDict.itemsds}}""$$rc6|jjS)z,Return an iterable of the BufferDict values.)rvaluesr!s rr3zBufferDict.valueshs}}##%%rc t|tjjs!t dt |j zt|ttfr|jD] \}}|||< y t|tjjr't|jD] \}}|||< y t|D]\}}t|tjjs0t dt|zdzt |j zt|dk(s2tdt|zdztt|zdz|d||d<y ) 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__rrr1Mappingsorted enumeratestrr ValueError)r rrrjps rr zBufferDict.updatelsj';??#;#;<8:>w-:P:PQ  g Z8 9&}} # V"S  # !8!8 9%gmmo6 # V"S  #"'* "1!![__%=%=>#>QGJccfjklfmfvfvv1v{$!!f%'568;CF DFWXqTQqT  "rcpg}|jjD]\}}djd|jD}|jsdnd|j d}dt j|d||d}|jd |zd z|zd j|}|S) Nxc32K|]}t|ywr)rA).0sizes r z(BufferDict.extra_repr..s?dD ?sz (GPU )zBuffer containing: [z of size ]z (z):  ) rr1joinrIis_cuda get_devicetorchtypenameappend)r child_lineskrDsize_str device_strparastrtmpstrs r extra_reprzBufferDict.extra_reprs MM'') :J!2LJ,U^^A->,?y S]R^^_`G   uqy507: ;  < ;' rctd)Nz BufferDict should not be called.) RuntimeError)r inputs r__call__zBufferDict.__call__s=>>r)NF)r bool)r= __module__ __qualname____doc__r rrrr"r&r(r*r.r%r1r3r r[r_ __classcell__)rs@rrrsU< !"F"*$ $%&%"N?rr) __future__rr8rrRtorch.nnrrr,rrrgs$## K?K?r