Update

app.py

@@ -39,7 +39,6 @@ if device.type == "cpu":
     pipe = MultiResPipeline.from_pretrained(model_id)
 else:
     pipe = MultiResPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
-pipe = pipe.to(device)
 pipe.scheduler = DPMSolverMultistepScheduler(
     beta_start=0.00085,
     beta_end=0.012,
@@ -53,6 +52,10 @@ pipe.scheduler = DPMSolverMultistepScheduler(
     lower_order_final=True,
 )
 string_to_param_dict = load_learned_concepts(pipe, "textual_inversion_outputs/")
+for k, v in list(string_to_param_dict.items()):
+    string_to_param_dict[k] = v.to(device)
+pipe.to(device)
+pipe.text_encoder.to(device)
 
 
 @spaces.GPU