@ -12,8 +12,8 @@ import io
import base64
from PIL import Image, PngImagePlugin
#from .pygmalion_helpers import get_full_prompt, num_tokens
from .llama_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
logger = logging.getLogger(__name__)