Spaces:
Build error
Build error
alielfilali01
commited on
Commit
•
475ade1
1
Parent(s):
7a90af6
Update app.py
Browse files
app.py
CHANGED
@@ -3,9 +3,6 @@
|
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, set_seed, pipeline
|
4 |
import gradio as gr
|
5 |
|
6 |
-
#import torch
|
7 |
-
#torch.set_default_tensor_type(torch.cuda.FloatTensor)
|
8 |
-
|
9 |
### need more GPU power to call better models !!!!!!
|
10 |
|
11 |
# model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp", use_cache=True) # 11B param
|
|
|
3 |
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM, set_seed, pipeline
|
4 |
import gradio as gr
|
5 |
|
|
|
|
|
|
|
6 |
### need more GPU power to call better models !!!!!!
|
7 |
|
8 |
# model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp", use_cache=True) # 11B param
|