Gopikanth123 commited on
Commit
72614ba
·
verified ·
1 Parent(s): a38fbf0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -23,12 +23,12 @@ quantization_config = BitsAndBytesConfig(
23
  )
24
 
25
  model = AutoModelForCausalLM.from_pretrained(
26
- "./results/checkpoint-639", # Replace with your model path or Hugging Face model name
27
  quantization_config=quantization_config,
28
  device_map="auto" # Automatically distribute across devices
29
  )
30
 
31
- tokenizer = AutoTokenizer.from_pretrained("./results/checkpoint-639/")
32
 
33
  # Create a text generation pipeline
34
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
23
  )
24
 
25
  model = AutoModelForCausalLM.from_pretrained(
26
+ "Gopikanth123/llama2_model", # Replace with your model path or Hugging Face model name
27
  quantization_config=quantization_config,
28
  device_map="auto" # Automatically distribute across devices
29
  )
30
 
31
+ tokenizer = AutoTokenizer.from_pretrained("Gopikanth123/llama2_model")
32
 
33
  # Create a text generation pipeline
34
  generator = pipeline("text-generation", model=model, tokenizer=tokenizer)