Hugging Face's logo

Spaces:
Mihaiii
/
backtrack_sampler_demo
Running on Zero

App Files Community
Mihaiii commited on
Commit
d0e140f
·
verified ·
1 Parent(s): fa9f7fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ model1 = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
20
  model2 = AutoModelForCausalLM.from_pretrained(model_name)
21
  device = torch.device('cuda')
22
 
23
- strategy = CreativeWritingStrategy(top_p_flat = 0.8, top_k_threshold_flat = 2, min_prob_second_highest = 0.2)
24
  provider = TransformersProvider(model2, tokenizer, device)
25
  creative_sampler = BacktrackSampler(strategy, provider)
26
 
 
20
  model2 = AutoModelForCausalLM.from_pretrained(model_name)
21
  device = torch.device('cuda')
22
 
23
+ strategy = CreativeWritingStrategy()
24
  provider = TransformersProvider(model2, tokenizer, device)
25
  creative_sampler = BacktrackSampler(strategy, provider)
26