wannaphong commited on
Commit
8ddee2d
·
1 Parent(s): 58d7ee7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,7 @@
2
  # warnings.filterwarnings("ignore")
3
  import gradio as gr
4
  import torch
5
- torch.set_num_threads()
6
- torch.set_num_interop_threads()
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
  from typing import Optional, Union, List, Dict, Any
9
  import random
@@ -133,6 +132,7 @@ model = AutoModelForCausalLM.from_pretrained(
133
  offload_folder="./",
134
  low_cpu_mem_usage=True,
135
  )
 
136
  tokenizer = AutoTokenizer.from_pretrained("facebook/xglm-7.5B")
137
 
138
 
 
2
  # warnings.filterwarnings("ignore")
3
  import gradio as gr
4
  import torch
5
+ torch.set_num_threads(1)
 
6
  from transformers import AutoModelForCausalLM, AutoTokenizer
7
  from typing import Optional, Union, List, Dict, Any
8
  import random
 
132
  offload_folder="./",
133
  low_cpu_mem_usage=True,
134
  )
135
+ model=torch.compile(model)
136
  tokenizer = AutoTokenizer.from_pretrained("facebook/xglm-7.5B")
137
 
138