diff --git a/fooocus_version.py b/fooocus_version.py index 3a5e5af..14c910b 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.5' +version = '2.0.8' diff --git a/modules/async_worker.py b/modules/async_worker.py index 76b5476..5f1e079 100644 --- a/modules/async_worker.py +++ b/modules/async_worker.py @@ -32,6 +32,7 @@ def worker(): print(e) def progressbar(number, text): + print(f'[Fooocus] {text}') outputs.append(['preview', (number, text, None)]) @torch.no_grad() diff --git a/modules/expansion.py b/modules/expansion.py index 7c15e0f..716d343 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -28,9 +28,14 @@ def remove_pattern(x, pattern): class FooocusExpansion: def __init__(self): + use_fp16 = model_management.should_use_fp16() + self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path) self.model = AutoModelForCausalLM.from_pretrained(fooocus_expansion_path) + if use_fp16: + self.model.half() + load_device = model_management.text_encoder_device() offload_device = model_management.text_encoder_offload_device() self.patcher = ModelPatcher(self.model, load_device=load_device, offload_device=offload_device) @@ -39,7 +44,7 @@ class FooocusExpansion: model=self.model, tokenizer=self.tokenizer, device='cpu', - torch_dtype=torch.float32) + torch_dtype=torch.float16 if use_fp16 else torch.float32) print(f'Fooocus Expansion engine loaded.')