|
@ -12,8 +12,8 @@ import io |
|
|
import base64 |
|
|
import base64 |
|
|
from PIL import Image, PngImagePlugin |
|
|
from PIL import Image, PngImagePlugin |
|
|
|
|
|
|
|
|
#from .pygmalion_helpers import get_full_prompt, num_tokens |
|
|
from .pygmalion_helpers import get_full_prompt, num_tokens |
|
|
from .llama_helpers import get_full_prompt, num_tokens |
|
|
#from .llama_helpers import get_full_prompt, num_tokens |
|
|
|
|
|
|
|
|
logger = logging.getLogger(__name__) |
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
|
|
|