Hugging Face's logo

Spaces:
LMSeed
/
Chat_with_Stu
Sleeping

App Files Community
Jia0603 commited on
Commit
39fec9a
·
verified ·
1 Parent(s): c097969

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
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
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
4
  import gradio as gr
5
 
6
+ MODEL_ID = "LMSeed/GPT2-Small-Distilled-900M" #"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