mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-04-04 14:17:28 +00:00
Make portable builds functional again after b3666e140d
This commit is contained in:
parent
6291e72129
commit
9d07d3a229
3 changed files with 44 additions and 39 deletions
|
|
@ -18,7 +18,6 @@ from modules.image_models import (
|
|||
)
|
||||
from modules.logging_colors import logger
|
||||
from modules.text_generation import stop_everything_event
|
||||
from modules.torch_utils import get_device
|
||||
from modules.utils import gradio
|
||||
|
||||
ASPECT_RATIOS = {
|
||||
|
|
@ -667,7 +666,7 @@ def generate(state):
|
|||
"""
|
||||
import torch
|
||||
|
||||
from modules.torch_utils import clear_torch_cache
|
||||
from modules.torch_utils import clear_torch_cache, get_device
|
||||
|
||||
clear_torch_cache()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue