Hugging Face's logo

Spaces:
andrew3279
/
Bloom_test
Build error

App Files Community
andrew3279 commited on
Commit
a108031
·
1 Parent(s): 83ac2ec

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"
12
 
13
  model = BloomForCausalLM.from_pretrained(model_name).to(device)
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).to(device)
14
  tokenizer = BloomTokenizerFast.from_pretrained(model_name)