Hugging Face's logo

Spaces:
Ghosthash
/
streamlit-higher
Sleeping

App Files Community
Ghosthash commited on
Commit
a240f76
·
verified ·
1 Parent(s): 40f49ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ model_path = "yubi12/autotrain-crypto-degen-tweet-generator-v1"
6
  tokenizer = AutoTokenizer.from_pretrained(model_path)
7
  model = AutoModelForCausalLM.from_pretrained(
8
  model_path,
9
- # device_map="auto",
10
- device = 0,
11
  torch_dtype='auto'
12
  ).eval()
13
 
 
6
  tokenizer = AutoTokenizer.from_pretrained(model_path)
7
  model = AutoModelForCausalLM.from_pretrained(
8
  model_path,
9
+ device_map="auto",
 
10
  torch_dtype='auto'
11
  ).eval()
12