MaxLSB commited on
Commit
ba13704
·
verified ·
1 Parent(s): 604d9cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import os
2
  import threading
3
  import gradio as gr
4
- import torch
5
  from transformers import (
6
  AutoModelForCausalLM,
7
  AutoTokenizer,
@@ -14,7 +13,6 @@ hf_token = os.environ["HUGGINGFACEHUB_API_TOKEN"]
14
  # Load tokenizer & model locally
15
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
16
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=hf_token)
17
- model = torch.compile(model)
18
  model.eval()
19
 
20
  def respond(
 
1
  import os
2
  import threading
3
  import gradio as gr
 
4
  from transformers import (
5
  AutoModelForCausalLM,
6
  AutoTokenizer,
 
13
  # Load tokenizer & model locally
14
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
15
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, token=hf_token)
 
16
  model.eval()
17
 
18
  def respond(