Hugging Face's logo

Spaces:
yasserrmd
/
Atlas-Chat
Runtime error

App Files Community
yasserrmd commited on
Commit
490d5f3
·
verified ·
1 Parent(s): d919dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -5,7 +5,6 @@ import spaces
5
 
6
  model_name = "MBZUAI-Paris/Atlas-Chat-27B"
7
  model = AutoModelForCausalLM.from_pretrained(
8
- "text-generation",
9
  model_name,
10
  model_kwargs={"torch_dtype": torch.bfloat16},
11
  device_map="auto"
 
5
 
6
  model_name = "MBZUAI-Paris/Atlas-Chat-27B"
7
  model = AutoModelForCausalLM.from_pretrained(
 
8
  model_name,
9
  model_kwargs={"torch_dtype": torch.bfloat16},
10
  device_map="auto"