fixing

app.py

@@ -70,9 +70,6 @@ def create_prompt_part(pipe, saes_dict, demo):
     @spaces.GPU
     def image_gen(prompt):
         lock.acquire()
-        pipe.to('cuda')
-        for sae in saes_dict.values:
-            sae.to('cuda')
         try:
             images, cache = pipe.run_with_cache(
                 prompt,