bypass cuda error

app.py

@@ -61,7 +61,10 @@ def get_pipeline(model_name):
     mdl = AutoModelForCausalLM.from_pretrained(
         model_name, weights_only=False, trust_remote_code=True
     )
-    mdl.to("cuda")
+    try:
+        mdl.to("cuda")
+    except Exception as e:
+        print(f'Error: {e}')
     return pipeline("text-generation", model=mdl, tokenizer=tok, device=0)
 
 @spaces.GPU