Only quantize flux T5 is also quantizing model. Load TE from original name and path if fine tuning.

This commit is contained in:
Jaret Burkett
2024-10-29 14:25:31 -06:00
parent 4747716867
commit 4aa19b5c1d
2 changed files with 11 additions and 34 deletions

View File

@@ -394,6 +394,8 @@ class TrainConfig:
class ModelConfig:
def __init__(self, **kwargs):
self.name_or_path: str = kwargs.get('name_or_path', None)
# name or path is updated on fine tuning. Keep a copy of the original
self.name_or_path_original: str = self.name_or_path
self.is_v2: bool = kwargs.get('is_v2', False)
self.is_xl: bool = kwargs.get('is_xl', False)
self.is_pixart: bool = kwargs.get('is_pixart', False)