Hugging Face's logo

Spaces:
andrew3279
/
Bloom_test
Build error

App Files Community
andrew3279 commited on
Commit
ada792a
·
1 Parent(s): d012d8e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import gradio as gr
8
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
  print(device)
10
 
11
- model_name = "bigscience/bloom-1b7"
12
 
13
  model = BloomForCausalLM.from_pretrained(model_name,)
14
  tokenizer = BloomTokenizerFast.from_pretrained(model_name)
 
8
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
9
  print(device)
10
 
11
+ model_name = "bigscience/bloom-1b1"
12
 
13
  model = BloomForCausalLM.from_pretrained(model_name,)
14
  tokenizer = BloomTokenizerFast.from_pretrained(model_name)