Type annotation fix (huggingface#840)

vwxyzjn · web-flow · commit 6c44096c7b8d · 2023-08-19T17:53:15.000+02:00
diff --git a/src/peft/tuners/lora.py b/src/peft/tuners/lora.py
@@ -104,7 +104,7 @@ class LoraConfig(PeftConfig):
             ),
         },
     )
-    layers_to_transform: Optional[Union[List, int]] = field(
+    layers_to_transform: Optional[Union[List[int], int]] = field(
         default=None,
         metadata={
             "help": "The layer indexes to transform, is this argument is specified, PEFT will transform only the layers indexes that are specified inside this list. If a single integer is passed, PEFT will transform only the layer at this index."

Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,7 @@ class LoraConfig(PeftConfig):`
`104`	`104`	`),`
`105`	`105`	`},`
`106`	`106`	`)`
`107`		`- layers_to_transform: Optional[Union[List, int]] = field(`
	`107`	`+ layers_to_transform: Optional[Union[List[int], int]] = field(`
`108`	`108`	`default=None,`
`109`	`109`	`metadata={`
`110`	`110`	`"help": "The layer indexes to transform, is this argument is specified, PEFT will transform only the layers indexes that are specified inside this list. If a single integer is passed, PEFT will transform only the layer at this index."`