From 1a088db0eb2a0bfef82643cc4d095742effacdf0 Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 13 Sep 2023 18:11:21 -0700 Subject: [PATCH] simplify --- modules/expansion.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/modules/expansion.py b/modules/expansion.py index f2dba0f..2efe881 100644 --- a/modules/expansion.py +++ b/modules/expansion.py @@ -27,12 +27,10 @@ def remove_pattern(x, pattern): class FooocusExpansion: def __init__(self): - 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 self.use_fp16: + if model_management.should_use_fp16(): self.model.half() load_device = model_management.text_encoder_device()