Hugging Face's logo

Spaces:
luna-code
/
luna-playground
Runtime error

App Files Community
1
terryyz commited on
Commit
d46f31c
·
verified ·
1 Parent(s): e90f65d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -180,7 +180,7 @@ def generate(
180
  model = model_map["CS Evo Prefix"]
181
  else:
182
  output = ""
183
- # model.to(device)
184
  input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
185
  # generated_ids = model.generate(**input_ids
186
  generated_ids = model.generate(**input_ids)#, **generate_kwargs)
 
180
  model = model_map["CS Evo Prefix"]
181
  else:
182
  output = ""
183
+ model.to(device)
184
  input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
185
  # generated_ids = model.generate(**input_ids
186
  generated_ids = model.generate(**input_ids)#, **generate_kwargs)