Hugging Face's logo

Spaces:
LMSeed
/
Chat_with_Stu
Sleeping

App Files Community
Jia0603 commited on
Commit
2fcc959
·
verified ·
1 Parent(s): 7a90291

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ import torch
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
4
  import gradio as gr
5
 
6
- MODEL_ID = "openai-community/gpt2"#"LMSeed/GPT2-small-distilled-100M_None_ppo-1000K-seed42"#"LMSeed/GPT2-small-distilled-100M"
 
7
 
8
  device = 0 if torch.cuda.is_available() else -1
9
 
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
4
  import gradio as gr
5
 
6
+ MODEL_ID = "LMSeed/GPT2-small-distilled-900M_None_ppo-1000K-seed42"
7
+ #"openai-community/gpt2"#"LMSeed/GPT2-small-distilled-100M"
8
 
9
  device = 0 if torch.cuda.is_available() else -1
10