yakine commited on
Commit
aa5d67a
·
verified ·
1 Parent(s): b6c092a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model_gpt2 = GPT2LMHeadModel.from_pretrained('gpt2')
20
  model_name = "meta-llama/Meta-Llama-3.1-8B"
21
  try:
22
  model_llama = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16,
23
- load_in_8bit=True. token = hf_token) # use device_map for automatic sharding
24
  except OSError as e:
25
  print(f"Error loading model: {e}")
26
 
 
20
  model_name = "meta-llama/Meta-Llama-3.1-8B"
21
  try:
22
  model_llama = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype=torch.float16,
23
+ load_in_8bit=True, token = hf_token) # use device_map for automatic sharding
24
  except OSError as e:
25
  print(f"Error loading model: {e}")
26