timep12345 commited on
Commit
764f570
1 Parent(s): ce18e89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -57,8 +57,7 @@ def url_changes(url, pages_to_visit, urls_to_scrape, repo_id):
57
  model = AutoModelForCausalLM.from_pretrained(
58
  MODEL,
59
  torch_dtype="auto",
60
- low_cpu_mem_usage=True,
61
- ).to(device=f"cuda", non_blocking=True)
62
  model.eval()
63
  pipe = pipeline(
64
  'text-generation',
 
57
  model = AutoModelForCausalLM.from_pretrained(
58
  MODEL,
59
  torch_dtype="auto",
60
+ )
 
61
  model.eval()
62
  pipe = pipeline(
63
  'text-generation',