rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Corrected requests import location for aidg.club support

Browse Source
This commit is contained in:
KoboldAI Dev
2021-05-16 05:37:38 -04:00
parent b05a73a04f
commit 47f1f7a85b
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
aiserver.py
View File

@ -9,6 +9,7 @@ from os import path, getcwd
import tkinter as tk import tkinter as tk
from tkinter import messagebox from tkinter import messagebox
import json import json
import requests
# KoboldAI # KoboldAI
import fileops import fileops
@ -239,8 +240,6 @@ if(not vars.model in ["InferKit", "Colab"]):
print("{0}OK! {1} pipeline created!{2}".format(colors.GREEN, vars.model, colors.END)) print("{0}OK! {1} pipeline created!{2}".format(colors.GREEN, vars.model, colors.END))
else: else:
# Import requests library for HTTPS calls
import requests
# If we're running Colab, we still need a tokenizer. # If we're running Colab, we still need a tokenizer.
if(vars.model == "Colab"): if(vars.model == "Colab"):
from transformers import GPT2Tokenizer from transformers import GPT2Tokenizer