mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
5d612ec095
- Rename old "model_management" directory to "model_management_OLD" in order to catch dangling references to original model manager. - Caught and fixed most dangling references (still checking) - Rename lora, textual_inversion and model_patcher modules - Introduce a RawModel base class to simplfy the Union returned by the model loaders. - Tidy up the model manager 2-related tests. Add useful fixtures, and a finalizer to the queue and installer fixtures that will stop the services and release threads.
15 lines
567 B
Python
15 lines
567 B
Python
"""Base class for 'Raw' models.
|
|
|
|
The RawModel class is the base class of LoRAModelRaw and TextualInversionModelRaw,
|
|
and is used for type checking of calls to the model patcher. Its main purpose
|
|
is to avoid a circular import issues when lora.py tries to import BaseModelType
|
|
from invokeai.backend.model_manager.config, and the latter tries to import LoRAModelRaw
|
|
from lora.py.
|
|
|
|
The term 'raw' was introduced to describe a wrapper around a torch.nn.Module
|
|
that adds additional methods and attributes.
|
|
"""
|
|
|
|
class RawModel:
|
|
"""Base class for 'Raw' model wrappers."""
|