Hugging Face's logo

Spaces:
Asahina2K
/
danbooru-tags-transformer
Runtime error

App Files Community
p1atdev commited on
Commit
63d1c50
•
1 Parent(s): 7651f6e

chore: remove optimum

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +1 -2
app.py CHANGED
@@ -4,7 +4,8 @@ import os
4
 
5
  import torch
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
- from optimum.onnxruntime import ORTModelForCausalLM
 
8
 
9
  from huggingface_hub import login
10
 
 
4
 
5
  import torch
6
  from transformers import AutoTokenizer, AutoModelForCausalLM
7
+
8
+ # from optimum.onnxruntime import ORTModelForCausalLM
9
 
10
  from huggingface_hub import login
11
 
requirements.txt CHANGED
@@ -1,3 +1,2 @@
1
  torch==2.1.0
2
- transformers==4.38.0
3
- optimum[onnxruntime]==1.17.1
 
1
  torch==2.1.0
2
+ transformers==4.38.0