mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

simplify

Browse Source
This commit is contained in:
lvmin 2023-09-13 18:11:21 -07:00
parent f7f548ff35
commit 1a088db0eb
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/expansion.py
View File

@ -27,12 +27,10 @@ def remove_pattern(x, pattern):
class FooocusExpansion: class FooocusExpansion:
def __init__(self): def __init__(self):
self.use_fp16 = model_management.should_use_fp16()
self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path) self.tokenizer = AutoTokenizer.from_pretrained(fooocus_expansion_path)
self.model = AutoModelForCausalLM.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() self.model.half()
load_device = model_management.text_encoder_device() load_device = model_management.text_encoder_device()