DanielWong76 commited on
Commit
2ea4f88
1 Parent(s): ed47413

Fixed repo being pulled from

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
 
6
  # Load the base model and LoRA adapter
7
  model_name = "unsloth/llama-3-8b-bnb-4bit" # Replace with your base model
8
- adapter_model_name = "your-hf-username/your-lora-model" # Replace with your LoRA adapter
9
 
10
  # Load the base model and tokenizer
11
  base_model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
 
5
 
6
  # Load the base model and LoRA adapter
7
  model_name = "unsloth/llama-3-8b-bnb-4bit" # Replace with your base model
8
+ adapter_model_name = "DanielWong76/lora_model1" # Replace with your LoRA adapter
9
 
10
  # Load the base model and tokenizer
11
  base_model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")