Tonic commited on
Commit
fd4f923
·
1 Parent(s): b1cb928

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -10,7 +10,6 @@ examples = [["How are you?"]]
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
12
  model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", device_map="auto", trust_remote_code=True).eval()
13
- model.generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) # Different generation length, top_p and other related super parameters can be specified.
14
 
15
  def predict(input, history=[]):
16
  # Check if input is not None and eos_token is not None
 
10
 
11
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True)
12
  model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen-7B-Chat", device_map="auto", trust_remote_code=True).eval()
 
13
 
14
  def predict(input, history=[]):
15
  # Check if input is not None and eos_token is not None