Gosula commited on
Commit
2f468d8
1 Parent(s): 55e2242

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_name, trust_remote_code=Tru
15
  tokenizer.pad_token = tokenizer.eos_token
16
  tokenizer.padding_side = "right"
17
 
 
18
  def QLoRA_Chatgpt(prompt):
19
  print(prompt)
20
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
 
15
  tokenizer.pad_token = tokenizer.eos_token
16
  tokenizer.padding_side = "right"
17
 
18
+
19
  def QLoRA_Chatgpt(prompt):
20
  print(prompt)
21
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)