Update app.py

app.py

@@ -89,9 +89,9 @@ refiner.scheduler=EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.
 #refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config, beta_schedule="scaled_linear")
 #refiner.scheduler = EulerAncestralDiscreteScheduler.from_config(refiner.scheduler.config)
 
-tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False, device_map='balanced')
+tokenizer = AutoTokenizer.from_pretrained(checkpoint, add_prefix_space=False)
 tokenizer.tokenizer_legacy=False
-model = AutoModelForCausalLM.from_pretrained(checkpoint, device_map='balanced')
+model = AutoModelForCausalLM.from_pretrained(checkpoint).to('cuda')
 #model = torch.compile(model)
 
 upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cuda:0"))
@@ -265,7 +265,8 @@ def infer(
     refiner.to(device=device, dtype=torch.bfloat16)
     refine = refiner(
             prompt=f"{enhanced_prompt_2}, high quality masterpiece, complex details",
-            negative_prompt = negative_prompt,
+            negative_prompt = negative_prompt_1,
+            negative_prompt_2 = negative_prompt_2,
             guidance_scale=7.5,
             num_inference_steps=num_inference_steps,
             image=sd_image,