preprocess algo is optimized and some code parts are refactored
This commit is contained in:
+1
-1
@@ -16,7 +16,7 @@ class PromptGenerator:
|
||||
[
|
||||
file
|
||||
for file in listdir(join(models_dir, "prompt_generators"))
|
||||
if isdir(join(join(models_dir, "prompt_generators"), file))
|
||||
if isdir(join(models_dir, "prompt_generators", file))
|
||||
],
|
||||
),
|
||||
"accelerate": (["enable", "disable"],),
|
||||
|
||||
Reference in New Issue
Block a user