Lifeinhockey commited on
Commit
575f020
·
verified ·
1 Parent(s): ea7103e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  # Загружаем токенизатор и модель
4
  # tokenizer = AutoTokenizer.from_pretrained("ai-forever/ruGPT-3.5-13B")
5
  # model = AutoModelForCausalLM.from_pretrained("ai-forever/ruGPT-3.5-13B")
6
- tokenizer = AutoTokenizer.from_pretrained("Gaivoronsky/ruGPT-3.5-13B-8bit")
7
- model = AutoModelForCausalLM.from_pretrained("Gaivoronsky/ruGPT-3.5-13B-8bit")
8
 
9
  prompt = "Привет! Как дела?"
10
  input_ids = tokenizer.encode(prompt, return_tensors="pt")
 
3
  # Загружаем токенизатор и модель
4
  # tokenizer = AutoTokenizer.from_pretrained("ai-forever/ruGPT-3.5-13B")
5
  # model = AutoModelForCausalLM.from_pretrained("ai-forever/ruGPT-3.5-13B")
6
+ tokenizer = AutoTokenizer.from_pretrained("ruGPT-3.5-13B-Kilusha ")
7
+ model = AutoModelForCausalLM.from_pretrained("ruGPT-3.5-13B-Kilusha ")
8
 
9
  prompt = "Привет! Как дела?"
10
  input_ids = tokenizer.encode(prompt, return_tensors="pt")