Hugging Face's logo

Spaces:
pszemraj
/
small-instruct-streaming
Sleeping

App Files Community
pszemraj commited on
Commit
35700da
•
1 Parent(s): 58acd65

use a new model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  import gradio as gr
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
13
 
14
- model_id = "BEE-spoke-data/tFINE-900m-e16-d32-instruct_2e"
15
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
16
  logging.info(f"Running on device:\t {torch_device}")
17
  logging.info(f"CPU threads:\t {torch.get_num_threads()}")
 
11
  import gradio as gr
12
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
13
 
14
+ model_id = "pszemraj/tFINE-850m-24x24-v0.5-instruct-L1"
15
  torch_device = "cuda" if torch.cuda.is_available() else "cpu"
16
  logging.info(f"Running on device:\t {torch_device}")
17
  logging.info(f"CPU threads:\t {torch.get_num_threads()}")