Tousifahamed commited on
Commit
c7c3e26
·
verified ·
1 Parent(s): 55e33aa

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +5 -2
app.py CHANGED
@@ -1,4 +1,7 @@
1
  import torch
 
 
 
2
  import torch.nn as nn
3
  import torch.quantization # <--- Use the older namespace for default_qconfig
4
  from transformers import AutoTokenizer
 
1
  import torch
2
+ print("PyTorch version:", torch.__version__)
3
+ print("Supported quantized engines:", torch.backends.quantized.supported_engines)
4
+
5
  import torch.nn as nn
6
  import torch.quantization # <--- Use the older namespace for default_qconfig
7
  from transformers import AutoTokenizer
requirements.txt CHANGED
@@ -1,3 +1,6 @@
1
- torch
 
 
 
2
  transformers
3
- gradio
 
1
+ --extra-index-url https://download.pytorch.org/whl/cpu
2
+ torch==2.0.1+cpu
3
+ torchvision==0.15.2+cpu
4
+ torchaudio==2.0.2+cpu
5
  transformers
6
+ gradio