alielfilali01 commited on
Commit
475ade1
1 Parent(s): 7a90af6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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