fix int vs float

2026-09-19 17:24:32 +02:00 · 2024-09-10 11:06:58 -04:00
parent ef12df3e98
commit c497321f7e
1 changed files with 4 additions and 4 deletions
@@ -44,11 +44,11 @@ class Script(scripts.Script):
        input_ids = self.tokenizer(input_text, return_tensors="pt").input_ids.to(devices.device)
        self.model = self.model.to(devices.device)
        kwargs = {
-            'max_length': max_length,
-            'num_return_sequences': num_return_sequences,
+            'max_length': int(max_length),
+            'num_return_sequences': int(num_return_sequences),
            'do_sample': True,
-            'temperature': temperature,
-            'repetition_penalty': repetition_penalty
+            'temperature': float(temperature),
+            'repetition_penalty': float(repetition_penalty),
        }
        try:
            outputs = self.model.generate(input_ids, **kwargs)