Hugging Face's logo

Spaces:
Pippoz
/
Hugging_Space
Runtime error

App Files Community
1
Pippoz commited on
Commit
fbed234
•
1 Parent(s): 741c239

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -3,7 +3,6 @@ import time
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
6
- @st.cache(allow_output_mutation=True)
7
  model = AutoModelForCausalLM.from_pretrained("facebook/opt-30b", torch_dtype=torch.float16).cuda()
8
  tokenizer = AutoTokenizer.from_pretrained("facebook/opt-30b", use_fast=False)
9
 
 
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
  import torch
5
 
 
6
  model = AutoModelForCausalLM.from_pretrained("facebook/opt-30b", torch_dtype=torch.float16).cuda()
7
  tokenizer = AutoTokenizer.from_pretrained("facebook/opt-30b", use_fast=False)
8