move to new backend - part 2

This commit is contained in:
layerdiffusion
2024-08-03 15:10:37 -07:00
parent 8a01b2c5db
commit 4add428e25
9 changed files with 61 additions and 69 deletions

View File

@@ -1,5 +1,5 @@
from modules.sd_hijack_clip import FrozenCLIPEmbedderWithCustomWords
from ldm_patched.modules import model_management
from backend import memory_management
from modules import sd_models
from modules.shared import opts
@@ -9,7 +9,7 @@ def move_clip_to_gpu():
print('Error: CLIP called before SD is loaded!')
return
model_management.load_model_gpu(sd_models.model_data.sd_model.forge_objects.clip.patcher)
memory_management.load_model_gpu(sd_models.model_data.sd_model.forge_objects.clip.patcher)
return