Hugging Face's logo

Spaces:
ClueAI
/
ChatYuan-large-v2
Runtime error

App Files Community
4
ClueAI commited on
Commit
cc2e6d4
•
1 Parent(s): a9d22e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
  import gradio as gr
3
  import clueai
 
4
  from transformers import T5Tokenizer, T5ForConditionalGeneration
5
  tokenizer = T5Tokenizer.from_pretrained("ClueAI/ChatYuan-large-v2")
6
  model = T5ForConditionalGeneration.from_pretrained("ClueAI/ChatYuan-large-v2").half()
 
1
  import os
2
  import gradio as gr
3
  import clueai
4
+ import torch
5
  from transformers import T5Tokenizer, T5ForConditionalGeneration
6
  tokenizer = T5Tokenizer.from_pretrained("ClueAI/ChatYuan-large-v2")
7
  model = T5ForConditionalGeneration.from_pretrained("ClueAI/ChatYuan-large-v2").half()